Add install variant for freqai

koosai · Jul 31, 2022 · cbb0535 · cbb0535
1 parent 61693f6
commit cbb0535
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/requirements-freqai.txt b/requirements-freqai.txt
@@ -3,7 +3,6 @@
 
 # Required for freqai
 scikit-learn==1.1.1
-scikit-optimize==0.9.0
 joblib==1.1.0
 catboost==1.0.4
 lightgbm==3.3.2
diff --git a/setup.py b/setup.py
@@ -12,6 +12,13 @@
     'progressbar2',
 ]
 
+freqai = [
+    'scikit-learn',
+    'joblib',
+    'catboost',
+    'lightgbm',
+]
+
 develop = [
     'coveralls',
     'flake8',
@@ -31,7 +38,7 @@
     'nbconvert',
 ]
 
-all_extra = plot + develop + jupyter + hyperopt
+all_extra = plot + develop + jupyter + hyperopt + freqai
 
 setup(
     tests_require=[
@@ -79,6 +86,7 @@
         'plot': plot,
         'jupyter': jupyter,
         'hyperopt': hyperopt,
+        'freqai': freqai,
         'all': all_extra,
     },
 )