From 876602372a353ac12f2aed0b55bf96747266d2f2 Mon Sep 17 00:00:00 2001 From: Tomaz Vieira Date: Fri, 24 Nov 2023 15:16:30 +0100 Subject: [PATCH] Fixes conftest --- tests/conftest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index a4efa28a..6a8366ae 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -11,8 +11,8 @@ from pytest import FixtureRequest, fixture os.environ["BIOIMAGEIO_COUNT_RDF_DOWNLOADS"] = "false" # disable tracking before bioimageio imports -from bioimageio.core import write_package from bioimageio.spec import __version__ as bioimageio_spec_version +from bioimageio.spec._package import save_bioimageio_package logger = logging.getLogger(__name__) warnings.warn(f"testing with bioimageio.spec {bioimageio_spec_version}") @@ -133,7 +133,7 @@ @fixture(scope="session") def model_packages(): - return MappingProxyType({name: write_package(MODEL_SOURCES[name]) for name in load_model_packages}) + return MappingProxyType({name: save_bioimageio_package(MODEL_SOURCES[name]) for name in load_model_packages}) @fixture(scope="session")