add torch to dependencies

radionets-project · Jul 18, 2024 · 6495516 · 6495516
1 parent 7602687
commit 6495516
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,6 +33,7 @@ requires-python = ">=3.10"
 dependencies = [
     "numpy",
     "astropy<=6.1.0",
+    "pytorch",
     "matplotlib",
     "ipython",
     "scipy",