add extras to setup.py for RL
This commit is contained in:
parent
cf10a76a2a
commit
017e476f49
11
setup.py
11
setup.py
@ -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,
|
||||
},
|
||||
)
|
||||
|
Loading…
Reference in New Issue
Block a user