Merge pull request #32 from capitalone/tazitoo-patch-1

Resolves #30 , Resolves #31 - Bump torch version to avoid vulns
capitalone · Jun 15, 2024 · 2cf7f08 · 2cf7f08
2 parents 7877ae0 + 1c66181
commit 2cf7f08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -5,7 +5,7 @@ seaborn
 pandas
 scikit-learn>=1.1.0
 captum
-torch>=1.13.1, < 2.0.0
+torch>=2.0.0
 pytorch_lightning>=1.6.5, < 2.0.0
 certifi>=2022.12.07
 werkzeug >= 2.2.3