refactor(runtime): Integrate metahyper closely

automl · Apr 25, 2024 · cc124c1 · cc124c1
1 parent cf34bdc
commit cc124c1
Show file tree

Hide file tree

Showing 35 changed files with 1,396 additions and 1,077 deletions.
diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
@@ -41,4 +41,4 @@ jobs:
 
       - name: Run pytest
         timeout-minutes: 15
-        run: poetry run pytest -m "all_examples or metahyper or neps_api or summary_csv"
+        run: poetry run pytest -m "all_examples or runtime or neps_api or summary_csv"
diff --git a/neps/api.py b/neps/api.py
@@ -1,6 +1,5 @@
 """API for the neps package.
 """
-
 from __future__ import annotations
 
 import logging
@@ -10,7 +9,8 @@
 
 import ConfigSpace as CS
 
-from .metahyper import instance_from_map, metahyper_run
+from neps.utils.common import instance_from_map
+from neps.runtime import launch_runtime
 from .optimizers import BaseOptimizer, SearcherMapping
 from .plot.tensorboard_eval import tblogger
 from .search_spaces.parameter import Parameter
@@ -271,7 +271,7 @@ def run(
     if development_stage_id is not None:
         root_directory = Path(root_directory) / f"dev_{development_stage_id}"
 
-    metahyper_run(
+    launch_runtime(
         run_pipeline,
         searcher_instance,
         searcher_info,

diff --git a/neps/metahyper/__init__.py b/neps/metahyper/__init__.py
diff --git a/neps/metahyper/_locker.py b/neps/metahyper/_locker.py