diff options
| -rw-r--r-- | models/model.py | 2 | ||||
| -rw-r--r-- | utils/dataset.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/models/model.py b/models/model.py index c6f6e57..7099b80 100644 --- a/models/model.py +++ b/models/model.py @@ -315,7 +315,7 @@ class Model: self, dataset_config: DatasetConfiguration, dataloader_config: DataloaderConfiguration, - ) -> tuple[DataLoader, dict[str: DataLoader]]: + ) -> tuple[DataLoader, dict[str, DataLoader]]: dataset_name = dataset_config.get('name', 'CASIA-B') if dataset_name == 'CASIA-B': gallery_dataset = self._parse_dataset_config( diff --git a/utils/dataset.py b/utils/dataset.py index ded9fd5..cd8b0f1 100644 --- a/utils/dataset.py +++ b/utils/dataset.py @@ -75,7 +75,7 @@ class CASIAB(data.Dataset): self.views: np.ndarray[np.str_] # Labels, classes, conditions and views in dataset, # set of three attributes above - self.metadata = dict[str, list[np.int64, str]] + self.metadata: dict[str, list[np.int64, str]] # Dictionaries for indexing frames and frame names by clip name # and chip path when cache is on |