From ad511be260d117ca1f2cf4ed05834a4f54af8af2 Mon Sep 17 00:00:00 2001
From: Felipe <fealho@gmail.com>
Date: Fri, 7 Jul 2023 21:00:17 -0700
Subject: [PATCH] Remove ==

---
 setup.py | 4 ++--
 tasks.py | 1 -
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 12de5238..db0abe7f 100644
--- a/setup.py
+++ b/setup.py
@@ -19,8 +19,8 @@
     "pandas>=1.5.0;python_version>'3.10'",
     "scikit-learn>=1.1.3,<2;python_version>='3.10'",
     "torch>=1.8.0;python_version<'3.10'",
-    "torch>=1.11.0;python_version=='3.10'",
-    "torch>=2.0.0;python_version>'3.10'",
+    "torch>=1.11.0;python_version>='3.10' and python_version<'3.11'",
+    "torch>=2.0.0;python_version>='3.11'",
     'rdt>=1.3.0,<2.0',
 ]
 
diff --git a/tasks.py b/tasks.py
index 89cce1c6..52e1a828 100644
--- a/tasks.py
+++ b/tasks.py
@@ -16,7 +16,6 @@
     '>': operator.gt,
     '<': operator.lt,
     '<=': operator.le,
-    '==': operator.eq
 }
 
 if not hasattr(inspect, 'getargspec'):