improve CondaEnv._get_version_pin

Author: FynnBe

bioimageio/spec/conda_env.py

@@ -50,8 +50,8 @@ def _ensure_valid_conda_env_name(cls, value: Optional[str]) -> Optional[str]:
     def wo_name(self):
         return self.model_construct(**{k: v for k, v in self if k != "name"})
 
-    def _get_version(self, package: str):
-        """Helper to return any verison pin for **package**
+    def _get_version_pin(self, package: str):
+        """Helper to return any version pin for **package**
 
         TODO: improve: interprete version pin and return structured information.
         """
@@ -62,6 +62,10 @@ def _get_version(self, package: str):
                         return p[len(package) :]
             elif d.startswith(package):
                 return d[len(package) :]
+            elif "::" in d and (d_wo_channel := d.split("::", 1)[-1]).startswith(
+                package
+            ):
+                return d_wo_channel[len(package) :]
 
     def get_pip_deps(self) -> List[str]:
         """Get the pip dependencies of this conda env."""

tests/test_get_conda_env.py

@@ -121,7 +121,8 @@ def test_get_default_pytorch_env():
     ]
     for p in ["torchvision", "torchaudio"]:
         versions[p] = [
-            env._get_version(p) for env in envs  # pyright: ignore[reportPrivateUsage]
+            env._get_version_pin(p)  # pyright: ignore[reportPrivateUsage]
+            for env in envs
         ]
 
     def assert_lt(p: str, i: int):