1 files changed, 12 insertions, 12 deletions
diff --git a/utils/configuration.py b/utils/configuration.py
index 579d2f2..f18dab0 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -1,4 +1,4 @@
-from typing import TypedDict, Optional, Union
+from typing import TypedDict, Optional, Union, Tuple, Dict
 
 from utils.dataset import ClipClasses, ClipConditions, ClipViews
 
@@ -19,35 +19,35 @@ class DatasetConfiguration(TypedDict):
     num_sampled_frames: int
     truncate_threshold: int
     discard_threshold: int
-    selector: Optional[dict[str, Union[ClipClasses, ClipConditions, ClipViews]]]
+    selector: Optional[Dict[str, Union[ClipClasses, ClipConditions, ClipViews]]]
     num_input_channels: int
-    frame_size: tuple[int, int]
+    frame_size: Tuple[int, int]
     cache_on: bool
 
 
 class DataloaderConfiguration(TypedDict):
-    batch_size: tuple[int, int]
+    batch_size: Tuple[int, int]
     num_workers: int
     pin_memory: bool
 
 
 class ModelHPConfiguration(TypedDict):
     ae_feature_channels: int
-    f_a_c_p_dims: tuple[int, int, int]
-    hpm_scales: tuple[int, ...]
+    f_a_c_p_dims: Tuple[int, int, int]
+    hpm_scales: Tuple[int, ...]
     hpm_use_avg_pool: bool
     hpm_use_max_pool: bool
     tfa_num_parts: int
     tfa_squeeze_ratio: int
-    embedding_dims: tuple[int]
+    embedding_dims: Tuple[int]
     triplet_is_hard: bool
     triplet_is_mean: bool
-    triplet_margins: tuple[float, float]
+    triplet_margins: Tuple[float, float]
 
 
 class SubOptimizerHPConfiguration(TypedDict):
     lr: int
-    betas: tuple[float, float]
+    betas: Tuple[float, float]
     eps: float
     weight_decay: float
     amsgrad: bool
@@ -55,7 +55,7 @@ class SubOptimizerHPConfiguration(TypedDict):
 
 class OptimizerHPConfiguration(TypedDict):
     lr: int
-    betas: tuple[float, float]
+    betas: Tuple[float, float]
     eps: float
     weight_decay: float
     amsgrad: bool
@@ -89,8 +89,8 @@ class ModelConfiguration(TypedDict):
     name: str
     restore_iter: int
     total_iter: int
-    restore_iters: tuple[int, ...]
-    total_iters: tuple[int, ...]
+    restore_iters: Tuple[int, ...]
+    total_iters: Tuple[int, ...]
 
 
 class Configuration(TypedDict):