fix: type error in main

MultimediaTechLab · Jan 23, 2025 · 8733d49 · 8733d49
1 parent 9de3de6
commit 8733d49
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/yolo/lazy.py b/yolo/lazy.py
@@ -3,6 +3,7 @@
 
 import hydra
 from lightning import Trainer
+from omegaconf.dictconfig import DictConfig
 
 # FIXME: messing with sys.path is a bad idea. Factor this out.
 project_root = Path(__file__).resolve().parent.parent
@@ -14,7 +15,7 @@
 
 
 @hydra.main(config_path="config", config_name="config", version_base=None)
-def main(cfg: Config):
+def main(cfg: DictConfig):
     callbacks, loggers, save_path = setup(cfg)
 
     trainer = Trainer(