add torch to dependencies

AllenInstitute · Jun 21, 2024 · a651425 · a651425
1 parent 37daad7
commit a651425
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -24,6 +24,7 @@ ssm @ git+https://github.com/lindermanlab/ssm
 statsmodels==0.14.0
 suite2p==0.12.1
 tensortools==0.4
+torch
 traitlets==5.6.0
 tqdm==4.66.2
 zarr==2.13.3