list pyspark in setup.py

isi-nlp · Aug 15, 2021 · 3087a58 · 3087a58
1 parent 8e7dd7a
commit 3087a58
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -43,7 +43,8 @@
         'torch >= 1.8.0',
         'sacremoses >= 0.0.45',
         'portalocker >= 2.0.0',
-        'torchtext >= 0.10.0'
+        'torchtext >= 0.10.0',
+        'pyspark >= 3.0.0'
     ],
     extra_requires={
         'big': ['pyspark >= 3.0'],