use BIOIMAGEIO_NO_CACHE in export_resource_package

Author: FynnBe

bioimageio/core/resource_io/io_.py

@@ -1,6 +1,7 @@
 import os
 import pathlib
 from copy import deepcopy
+from tempfile import TemporaryDirectory
 from typing import Dict, Optional, Sequence, Union
 from zipfile import ZIP_DEFLATED, ZipFile
 
@@ -10,7 +11,12 @@
 from bioimageio.core.resource_io.nodes import ResourceDescription
 from bioimageio.spec import load_raw_resource_description
 from bioimageio.spec.shared import raw_nodes
-from bioimageio.spec.shared.common import BIOIMAGEIO_CACHE_PATH, get_class_name_from_type
+from bioimageio.spec.shared.common import (
+    BIOIMAGEIO_CACHE_PATH,
+    BIOIMAGEIO_NO_CACHE,
+    get_class_name_from_type,
+    no_cache_tmp_list,
+)
 from bioimageio.spec.shared.raw_nodes import ResourceDescription as RawResourceDescription
 from . import nodes
 from .utils import resolve_raw_resource_description, resolve_source
@@ -134,10 +140,15 @@ def _get_package_base_name(raw_rd: RawResourceDescription, weights_priority_orde
 
 
 def _get_tmp_package_path(raw_rd: RawResourceDescription, weights_priority_order: Optional[Sequence[str]]):
-    package_file_name = _get_package_base_name(raw_rd, weights_priority_order)
+    if BIOIMAGEIO_NO_CACHE:
+        tmp_dir = TemporaryDirectory()
+        no_cache_tmp_list.append(tmp_dir)
+        return pathlib.Path(tmp_dir.name)
 
+    package_file_name = _get_package_base_name(raw_rd, weights_priority_order)
     cache_folder = BIOIMAGEIO_CACHE_PATH / "packages"
     cache_folder.mkdir(exist_ok=True, parents=True)
+
     package_path = (cache_folder / package_file_name).with_suffix(".zip")
     max_cached_packages_with_same_name = 100
     for p in range(max_cached_packages_with_same_name):