Add pytorch-3dunet dependency

ilastik · Dec 9, 2024 · f65ae9d · f65ae9d
1 parent 0cb3535
commit f65ae9d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/environment.yml b/environment.yml
@@ -9,7 +9,7 @@ dependencies:
   # - bioimage.core via submodule
   - python 3.9.*
   - numpy >=1.21,<2
-  - grpcio=1.44 # protobuf 5 requires protoc version > 3.19.0 that requires grpcio >= 1.44
+  - grpcio=1.49.1 # protobuf 5 requires protoc version > 3.19.0 that requires grpcio >= 1.44
   - marshmallow-union
   - marshmallow=3.12.*
   - marshmallow-jsonschema
@@ -20,6 +20,7 @@ dependencies:
   - scipy
   - typing-extensions
   - xarray
+  - pytorch-3dunet
   - setuptools
   - pip
 

diff --git a/setup.py b/setup.py
@@ -33,6 +33,7 @@
         "numpy<2",  # pytorch 2.2.2-py3.9_0 for macos is compiled with numpy 1.*
         "protobuf",
         "pydantic>=2.7.0,<2.10",
+        "pytorch-3dunet",
         "pyyaml",
         "xarray",
     ],