Some type hint fixes

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