add extras to setup.py for RL

This commit is contained in:
robcaulk 2022-10-05 17:20:40 +02:00
parent cf10a76a2a
commit 017e476f49

View File

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