Fixed list type casting bug (#28)

* Fixed list type casting bug * Fromatter applied to the trainer
fcakyon · Jan 8, 2023 · 6b69b5e · 6b69b5e
1 parent 7578afc
commit 6b69b5e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/video_transformers/trainer.py b/video_transformers/trainer.py
@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Dict, Union
+from typing import Dict, List, Union
 
 import numpy as np
 import torch
@@ -30,7 +30,7 @@ def __init__(
         mixed_precision: str = "no",
         output_dir: str = "runs",
         seed: int = 42,
-        trackers: list[GeneralTracker] = None,
+        trackers: List[GeneralTracker] = None,
         checkpoint_save: bool = True,
         checkpoint_save_interval: int = 1,
         checkpoint_save_policy: str = "epoch",