adapt SearcherConfigs to the new dict design of optimizers

automl · Jun 17, 2024 · 736b58a · 736b58a
1 parent f23c882
commit 736b58a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neps/optimizers/info.py b/neps/optimizers/info.py
@@ -56,7 +56,7 @@ def get_available_algorithms() -> list[str]:
                 file_path = os.path.join(folder_path, filename)
                 with open(file_path) as file:
                     searcher_config = yaml.safe_load(file)
-                    algorithm = searcher_config["searcher_init"].get("algorithm")
+                    algorithm = searcher_config.get("strategy")
                     if algorithm:
                         prev_algorithms.add(algorithm)
 
@@ -81,7 +81,7 @@ def get_searcher_from_algorithm(algorithm: str) -> list[str]:
                 file_path = os.path.join(folder_path, filename)
                 with open(file_path) as file:
                     searcher_config = yaml.safe_load(file)
-                    if searcher_config["searcher_init"].get("algorithm") == algorithm:
+                    if searcher_config.get("strategy") == algorithm:
                         searchers.append(os.path.splitext(filename)[0])
 
         return searchers