Defining ARBITRARY_SIZE

In case it is not defined, it defined it
MouseLand · Jan 7, 2025 · 4896dd4 · 4896dd4
1 parent 100081a
commit 4896dd4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cellpose/export.py b/cellpose/export.py
@@ -79,6 +79,12 @@
     Version,
     WeightsDescr,
 )
+# Define ARBITRARY_SIZE if it is not available in the module
+try:
+    from bioimageio.spec.model.v0_5 import ARBITRARY_SIZE
+except ImportError:
+    ARBITRARY_SIZE = 1  # or set to a default value if appropriate
+
 from bioimageio.spec.common import HttpUrl
 from bioimageio.spec import save_bioimageio_package
 from bioimageio.core import test_model