(chore): add torch for testing

theislab · Feb 26, 2024 · 133ed51 · 133ed51
1 parent 6e85255
commit 133ed51
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -66,6 +66,7 @@ dependencies = [
 neural = [
     "optax>=0.1.7",
     "flax>=0.7.1",
+    "torch"
 ]
 
 spatial = [