Skip to content

Commit

Permalink
fix type hints
Browse files Browse the repository at this point in the history
  • Loading branch information
pmeier committed Aug 24, 2020
1 parent 8a44845 commit c3c7805
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchvision/datasets/places365.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ def process(line: str) -> Tuple[str, int]:
images = [process(line) for line in fh]

_, targets = zip(*images)
return images, targets
return images, list(targets)

def download_devkit(self) -> None:
file, md5 = self._DEVKIT_META["challenge" if self.split == "train-challenge" else "standard"]
Expand Down

0 comments on commit c3c7805

Please sign in to comment.