add missing annotation

pytorch · Apr 6, 2022 · 5062a32 · 5062a32
1 parent 8e7987a
commit 5062a32
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torchvision/prototype/datasets/_builtin/pcam.py b/torchvision/prototype/datasets/_builtin/pcam.py
@@ -2,7 +2,6 @@
 import pathlib
 from collections import namedtuple
 from typing import Any, Dict, List, Optional, Tuple, Iterator, Union
-from unicodedata import category
 
 from torchdata.datapipes.iter import IterDataPipe, Mapper, Zipper
 from torchvision.prototype import features
@@ -130,5 +129,5 @@ def _datapipe(self, resource_dps: List[IterDataPipe]) -> IterDataPipe[Dict[str,
         dp = hint_sharding(dp)
         return Mapper(dp, self._prepare_sample)
 
-    def __len__(self):
+    def __len__(self) -> int:
         return 262_144 if self._split == "train" else 32_768