Add install variant for freqai

This commit is contained in:
Matthias 2022-07-31 15:10:01 +02:00
parent 61693f6c8b
commit cbb05354a8
2 changed files with 9 additions and 2 deletions

View File

@ -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

View File

@ -12,6 +12,13 @@ hyperopt = [
'progressbar2',
]
freqai = [
'scikit-learn',
'joblib',
'catboost',
'lightgbm',
]
develop = [
'coveralls',
'flake8',
@ -31,7 +38,7 @@ jupyter = [
'nbconvert',
]
all_extra = plot + develop + jupyter + hyperopt
all_extra = plot + develop + jupyter + hyperopt + freqai
setup(
tests_require=[
@ -79,6 +86,7 @@ setup(
'plot': plot,
'jupyter': jupyter,
'hyperopt': hyperopt,
'freqai': freqai,
'all': all_extra,
},
)