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'):