From 9bd6908ed77410e2efa439d26ebd1f9b60519f53 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 12 Jan 2021 11:43:11 +0800 Subject: Some type hint fixes --- models/model.py | 2 +- 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 -- cgit v1.2.3