Catch exception of initialize_config_dir in hydra

2024-08-05 11:39:47 +09:00 · 2024-08-05 11:39:47 +09:00 · e54e285072
parent 28fe6a7a0f
commit e54e285072
1 changed files with 4 additions and 1 deletions
--- a/sam2/init.py
+++ b/sam2/init.py
@ -14,5 +14,8 @@ warnings.filterwarnings("ignore", category=UserWarning, module="sam2")
 inpa_basedir = os.path.abspath(os.path.normpath(os.path.join(os.path.dirname(__file__), "..")))
 configs_path = os.path.join(inpa_basedir, "sam2_configs")

-initialize_config_dir(configs_path, version_base="1.2")
+try:
+    initialize_config_dir(configs_path, version_base="1.2")
+except TypeError:
+    initialize_config_dir(configs_path)
 # initialize_config_module("sam2_configs", version_base="1.2")