From d0edaf066d2c7069bce9af4102ec4c75ee4e2911 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Thu, 18 Feb 2021 20:57:37 +0800
Subject: Type hint fixes

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

(limited to 'models/model.py')

diff --git a/models/model.py b/models/model.py
index a3b6d3a..4bf8dbc 100644
--- a/models/model.py
+++ b/models/model.py
@@ -279,11 +279,11 @@ class Model:
     def transform(
             self,
             iters: Tuple[int],
-            dataset_config: DatasetConfiguration,
+            dataset_config: Dict,
             dataset_selectors: Dict[
                 str, Dict[str, Union[ClipClasses, ClipConditions, ClipViews]]
             ],
-            dataloader_config: DataloaderConfiguration
+            dataloader_config: Dict
     ):
         self.is_train = False
         # Split gallery and probe dataset
-- 
cgit v1.2.3