From 4ee31c0ec4038f9af46959248b90d31569b473d1 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Tue, 29 Dec 2020 20:59:40 +0800
Subject: Add type hint for new label (numpy.int64)

---
 utils/dataset.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/utils/dataset.py b/utils/dataset.py
index 050ac03..bb4d762 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[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
@@ -167,7 +167,10 @@ class CASIAB(data.Dataset):
     def __len__(self) -> int:
         return len(self.labels)
 
-    def __getitem__(self, index: int) -> dict[str, Union[str, torch.Tensor]]:
+    def __getitem__(
+            self,
+            index: int
+    ) -> dict[str, Union[np.int64, str, torch.Tensor]]:
         label = self.labels[index]
         condition = self.conditions[index]
         view = self.views[index]
-- 
cgit v1.2.3