diff --git a/neps_examples/__init__.py b/neps_examples/__init__.py new file mode 100644 index 00000000..bfc8f6aa --- /dev/null +++ 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 index 721365b2..abdd10c5 100644 --- 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]