diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-18 20:57:37 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-18 20:57:37 +0800 |
| commit | d0edaf066d2c7069bce9af4102ec4c75ee4e2911 (patch) | |
| tree | ec6499b1c9c56c6ac84d42e95be02fa1e1b72413 /models | |
| parent | 434f0e706649c68125c782060926ba39307c6cf1 (diff) | |
Type hint fixes
Diffstat (limited to 'models')
| -rw-r--r-- | models/model.py | 4 |
1 files changed, 2 insertions, 2 deletions
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 |