Add install variant for freqai
This commit is contained in:
parent
61693f6c8b
commit
cbb05354a8
@ -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
|
||||
|
10
setup.py
10
setup.py
@ -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,
|
||||
},
|
||||
)
|
||||
|
Loading…
Reference in New Issue
Block a user