lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 10:21:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

add extras to setup.py for RL

Browse Source
This commit is contained in:
robcaulk 2022-10-05 17:20:40 +02:00
parent cf10a76a2a
commit 017e476f49
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
setup.py
View File

@ -15,6 +15,14 @@ freqai = [
'scikit-learn', 'scikit-learn',
'catboost; platform_machine != "aarch64"', 'catboost; platform_machine != "aarch64"',
'lightgbm', 'lightgbm',
'xgboost'
]
freqai_rl = [
'torch',
'stable-baselines3',
'gym==0.21',
'sb3-contrib'
] ]
develop = [ develop = [
@ -36,7 +44,7 @@ jupyter = [
'nbconvert', 'nbconvert',
] ]
all_extra = plot + develop + jupyter + hyperopt + freqai all_extra = plot + develop + jupyter + hyperopt + freqai + freqai_rl
setup( setup(
tests_require=[ tests_require=[
@ -90,6 +98,7 @@ setup(
'jupyter': jupyter, 'jupyter': jupyter,
'hyperopt': hyperopt, 'hyperopt': hyperopt,
'freqai': freqai, 'freqai': freqai,
'freqai_rl': freqai_rl,
'all': all_extra, 'all': all_extra,
}, },
) )