Move examples lists from test to neps_examples

automl · Jul 1, 2024 · 14bec8e · 14bec8e
1 parent 7f0340f
commit 14bec8e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 17 deletions.
diff --git a/neps_examples/__init__.py b/neps_examples/__init__.py
@@ -0,0 +1,22 @@
+all_main_examples = {  # Used for printing in python -m neps_examples
+    "basic_usage": ["analyse", "architecture", "architecture_and_hyperparameters", "hpo_usage_example"],
+    "convenience": ["logging_additional_info", "neps_tblogger_tutorial", "neps_x_lightning", "working_directory_per_pipeline"],
+    "efficiency": ["expert_priors_for_hyperparameters", "multi_fidelity", "multi_fidelity_and_expert_priors"],
+}
+
+core_examples = [  # Run locally and on github actions
+    "basic_usage/hyperparameters",  # NOTE: This needs to be first for some tests to work
+    "basic_usage/analyse",
+    "experimental/expert_priors_for_architecture_and_hyperparameters",
+    "efficiency/multi_fidelity",
+]
+
+ci_examples = [  # Run on github actions
+    "basic_usage/architecture_and_hyperparameters",
+    "experimental/hierarchical_architecture",
+    "efficiency/expert_priors_for_hyperparameters",
+    "experimental/hierarchical_architecture_hierarchical_GP",
+    "convenience/logging_additional_info",
+    "convenience/working_directory_per_pipeline",
+    "convenience/neps_tblogger_tutorial",
+]
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -5,6 +5,8 @@
 
 import pytest
 
+from neps_examples import ci_examples, core_examples
+
 
 @pytest.fixture(autouse=True)
 def use_tmpdir(tmp_path, request):
@@ -24,23 +26,6 @@ def no_logs_gte_error(caplog):
     assert not errors
 
 
-core_examples = [  # Run locally and on github actions
-    "basic_usage/hyperparameters",  # NOTE: This needs to be first for the test below...
-    "basic_usage/analyse",
-    "experimental/expert_priors_for_architecture_and_hyperparameters",
-    "efficiency/multi_fidelity",
-]
-ci_examples = [  # Run on github actions
-    "basic_usage/architecture_and_hyperparameters",
-    "experimental/hierarchical_architecture",
-    "efficiency/expert_priors_for_hyperparameters",
-    "experimental/hierarchical_architecture_hierarchical_GP",
-    "convenience/logging_additional_info",
-    "convenience/working_directory_per_pipeline",
-    "convenience/neps_tblogger_tutorial",
-]
-
-
 examples_folder = Path(__file__, "..", "..", "neps_examples").resolve()
 core_examples_scripts = [examples_folder / f"{example}.py" for example in core_examples]
 ci_examples_scripts = [examples_folder / f"{example}.py" for example in ci_examples]