torch specify

polio-nanopore · Sep 12, 2024 · 36f1591 · 36f1591
1 parent 4dfe0a7
commit 36f1591
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -27,7 +27,8 @@
             "biopython",
             "medaka==2.0.0",
             "numpy",
-            "scipy"
+            "scipy",
+            "torch==2.4.1"
       ],
       description='piranha: Poliovirus Investigation Resource Automating Nanopore Haplotype Analysis',
       url='https://github.com/polio-nanopore/piranha',