diff --git a/neps/optimizers/info.py b/neps/optimizers/info.py index 2ed03f30..7088f341 100644 --- 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