From 45067321993deb8209161cd7e0b3a3d046a80be4 Mon Sep 17 00:00:00 2001
From: RichieHakim <RichHakim@gmail.com>
Date: Thu, 18 Apr 2024 16:19:36 -0400
Subject: [PATCH] Add opt_einsum to requirements.txt and setup.py

---
 requirements.txt | 2 +-
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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',