diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index e0451052..0ffcd162 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -1,4 +1,4 @@ -pytest_plugins = ["fixtures.neuron.service", "fixtures.neuron.model"] +pytest_plugins = ["fixtures.neuron.service", "fixtures.neuron.export_models"] # ruff: noqa: E402 from _pytest.fixtures import SubRequest import requests diff --git a/integration-tests/fixtures/neuron/model.py b/integration-tests/fixtures/neuron/export_models.py similarity index 97% rename from integration-tests/fixtures/neuron/model.py rename to integration-tests/fixtures/neuron/export_models.py index 2d58351c..a49642da 100644 --- a/integration-tests/fixtures/neuron/model.py +++ b/integration-tests/fixtures/neuron/export_models.py @@ -216,6 +216,11 @@ def maybe_export_model(config_name, model_config): return neuron_model_id +def maybe_export_models(): + for config_name, model_config in MODEL_CONFIGURATIONS.items(): + maybe_export_model(config_name, model_config) + + @pytest.fixture(scope="session", params=MODEL_CONFIGURATIONS.keys()) def neuron_model_config(request): """Expose a pre-trained neuron model @@ -262,3 +267,7 @@ def neuron_model_config(request): @pytest.fixture(scope="module") def neuron_model_path(neuron_model_config): yield neuron_model_config["neuron_model_path"] + + +if __name__ == "__main__": + maybe_export_models()