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',
|
'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,
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user