From 7f9aad73a4dd6e47b3d411149f396b85f4d11283 Mon Sep 17 00:00:00 2001
From: William Patton <wllmpttn24@gmail.com>
Date: Wed, 19 Feb 2025 11:38:58 -0800
Subject: [PATCH] Saving to bioimageio modelzoo is not yet supported for python
 3.10 and below

---
 dacapo/experiments/run_config.py       |  6 ++++++
 tests/components/test_architectures.py | 14 +++++++++++++-
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/dacapo/experiments/run_config.py b/dacapo/experiments/run_config.py
index 71290127..cfaa0b87 100644
--- a/dacapo/experiments/run_config.py
+++ b/dacapo/experiments/run_config.py
@@ -17,6 +17,8 @@
 from .validation_scores import ValidationScores
 from .model import Model
 
+import sys
+
 from bioimageio.core import test_model
 from bioimageio.spec import save_bioimageio_package
 from bioimageio.spec.model.v0_5 import (
@@ -455,6 +457,10 @@ def save_bioimage_io_model(
 
             weights_path = tmp / "model.pth"
             torch.save(self.model.state_dict(), weights_path)
+            if sys.version_info[1] < 11:
+                raise RuntimeError(
+                    "Saving to bioimageio modelzoo format is not implemented for Python versions < 3.11"
+                )
             with open(weights_path, "rb", buffering=0) as f:
                 weights_hash = hashlib.file_digest(f, "sha256").hexdigest()
 
diff --git a/tests/components/test_architectures.py b/tests/components/test_architectures.py
index 3f84d1dd..b5e3c2e4 100644
--- a/tests/components/test_architectures.py
+++ b/tests/components/test_architectures.py
@@ -32,6 +32,8 @@
 
 import pytest
 
+import sys
+
 
 def build_test_architecture_config(
     data_dims: int,
@@ -136,7 +138,17 @@ def build_test_architecture_config(
 @pytest.mark.parametrize("batch_norm", [True, False])
 @pytest.mark.parametrize("use_attention", [True, False])
 @pytest.mark.parametrize("padding", ["valid", "same"])
-@pytest.mark.parametrize("source", ["config", "module", "bioimage_modelzoo"])
+@pytest.mark.parametrize(
+    "source",
+    [
+        "config",
+        "module",
+        pytest.param(
+            "bioimage_modelzoo",
+            marks=pytest.mark.skipif(sys.version_info[1] < 11, reason="skip this one"),
+        ),
+    ],
+)
 def test_architectures(
     data_dims,
     channels,