diff --git a/requirements.txt b/requirements.txt index 938bb65..f491d2e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,6 +8,7 @@ natsort numba numpy opencv_contrib_python +opt_einsum optuna optuna_integration pandas @@ -31,7 +32,6 @@ torchaudio ## Optional / Advanced tables -opt_einsum # rolling_quantiles pulp spconv diff --git a/setup.py b/setup.py index 439e5e8..80de637 100644 --- a/setup.py +++ b/setup.py @@ -65,6 +65,7 @@ def read_requirements(): 'tqdm', 'h5py', 'opencv_contrib_python', + 'opt_einsum', 'optuna', 'optuna_integration', 'sparse', @@ -87,7 +88,6 @@ def read_requirements(): deps_advanced = {dep: deps_all_dict[dep] for dep in [ 'tables', - 'opt_einsum', # 'rolling_quantiles', 'pulp', 'spconv',