Add extra condition to requirements

stylecloud is not supported in Python 2, therefore, PySS3 will not support word cloud generation in Python 2.
sergioburdisso · Jun 12, 2021 · d9fe65d · d9fe65d
1 parent 61e75d5
commit d9fe65d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -9,4 +9,4 @@ scikit-learn[alldeps]>=0.20
 tqdm>=4.8.4
 matplotlib
 iterative-stratification
-stylecloud
+stylecloud; python_version >= '3'
diff --git a/requirements.txt b/requirements.txt
@@ -4,4 +4,4 @@ scikit-learn[alldeps]>=0.20
 tqdm>=4.8.4
 matplotlib
 iterative-stratification
-stylecloud
+stylecloud; python_version >= '3'
diff --git a/setup.py b/setup.py
@@ -72,4 +72,7 @@
                          'tqdm>=4.8.4',
                          'matplotlib',
                          'iterative-stratification'],
+          extras_require={
+              ':python_version >== "3"': ['stylecloud'],
+          },
           entry_points={'console_scripts': ['pyss3=pyss3.cmd_line:main']})