From 61f79339fd0168bc200117daf0459764b93d0acb Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Tue, 12 Jan 2021 14:26:03 +0800
Subject: Some type hint fixes

---
 models/model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'models')

diff --git a/models/model.py b/models/model.py
index 617966f..af37bb2 100644
--- a/models/model.py
+++ b/models/model.py
@@ -182,11 +182,11 @@ class Model:
     def predict_all(
             self,
             iter_: int,
-            dataset_config: dict,
+            dataset_config: Dict,
             dataset_selectors: Dict[
                 str, Dict[str, Union[ClipClasses, ClipConditions, ClipViews]]
             ],
-            dataloader_config: dict,
+            dataloader_config: Dict,
     ) -> Dict[str, torch.Tensor]:
         self.is_train = False
         # Split gallery and probe dataset
-- 
cgit v1.2.3