needs pytorch

bioimage-io · Nov 16, 2023 · 8263846 · 8263846
1 parent c9ee975
commit 8263846
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,7 @@ python-dateutil = "^2.8.2"
 annotated-types = "^0.6.0"
 typing-extensions = "^4.8.0"
 lxml = "^4.9.3"
+torch = "^2.1.1"
 
 [tool.poetry.group.dev.dependencies]
 pytest-cov = "^4.1.0"