Fix debug config

octo-models · Dec 13, 2023 · b6d504e · b6d504e
1 parent 44e6c63
commit b6d504e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/debug_config.py b/tests/debug_config.py
@@ -1,7 +1,12 @@
 from copy import deepcopy
 
 from ml_collections import ConfigDict
-from scripts.configs.config import get_config as get_base_config
+import imp
+import os
+
+get_base_config = imp.load_source(
+    "config", os.path.join(os.path.dirname(__file__), "../scripts/configs/config.py")
+).get_config
 
 
 def update_config(config: ConfigDict, **kwargs):