CHORE: Update requirements.txt to include torch and torchvision, and …

…comment out pyspark
MKaczkow · Feb 11, 2025 · c8404a7 · c8404a7
1 parent a8f1eac
commit c8404a7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,8 @@
 # for torch and torchvision use this (as of 2024-04-18 and Win)
 # pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118
 
+torch
+torchvision
 pytorch-lightning
 jupyter
 scikit-learn
@@ -11,7 +13,7 @@ xgboost
 ipykernel
 ipympl
 statsmodels
-pyspark
+# pyspark
 wordcloud
 nltk
 altair