diff --git a/setup.py b/setup.py index 304567bcc..894388554 100644 --- a/setup.py +++ b/setup.py @@ -25,6 +25,11 @@ freqai_rl = [ 'sb3-contrib' ] +hdf5 = [ + 'tables', + 'blosc', +] + develop = [ 'coveralls', 'flake8', @@ -44,7 +49,7 @@ jupyter = [ 'nbconvert', ] -all_extra = plot + develop + jupyter + hyperopt + freqai + freqai_rl +all_extra = plot + develop + jupyter + hyperopt + hdf5 + freqai + freqai_rl setup( tests_require=[ @@ -78,8 +83,6 @@ setup( 'prompt-toolkit', 'numpy', 'pandas', - 'tables', - 'blosc', 'joblib>=1.2.0', 'pyarrow; platform_machine != "armv7l"', 'fastapi', @@ -97,6 +100,7 @@ setup( 'plot': plot, 'jupyter': jupyter, 'hyperopt': hyperopt, + 'hdf5': hdf5, 'freqai': freqai, 'freqai_rl': freqai_rl, 'all': all_extra,