fix: 🐛 erros when "VAL" and "Test" are not in cfg

GestaltCogTeam · Dec 11, 2023 · a23494e · a23494e
1 parent e1e8c54
commit a23494e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/basicts/runners/base_runner.py b/basicts/runners/base_runner.py
@@ -26,9 +26,9 @@ def __init__(self, cfg: dict):
         super().__init__(cfg)
 
         # validate every `val_interval` epoch
-        self.val_interval = cfg["VAL"].get("INTERVAL", 1)
+        self.val_interval = cfg["VAL"].get("INTERVAL", 1) if hasattr(cfg, "VAL") else None
         # test every `test_interval` epoch
-        self.test_interval = cfg["TEST"].get("INTERVAL", 1)
+        self.test_interval = cfg["TEST"].get("INTERVAL", 1) if hasattr(cfg, "TEST") else None
 
         # declare data loader
         self.train_data_loader = None