Add opt_einsum to requirements.txt and setup.py

RichieHakim · Apr 18, 2024 · 4506732 · 4506732
1 parent c1b49bd
commit 4506732
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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',