Skip to content

Commit

Permalink
modifying typing
Browse files Browse the repository at this point in the history
  • Loading branch information
prabhuteja12 committed Oct 9, 2023
1 parent 2974322 commit 520c3ab
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/renate/benchmark/datasets/vision_datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -569,7 +569,7 @@ def setup(self) -> None:
self._train_data, self._val_data = self._split_train_val_data(train_data)
self._test_data = ImageDataset(*self._parse_file_lists("test"))

def _parse_file_lists(self, stage: str) -> Tuple[str, int]:
def _parse_file_lists(self, stage: str) -> Tuple[List[str], List[int]]:
data_id = self.data_id if stage == "train" else -1
idx_list = self._LUP[self._scenario][0][data_id]
X = [os.path.join(self._complete_data_path, self._paths[idx]) for idx in idx_list]
Expand Down

0 comments on commit 520c3ab

Please sign in to comment.