Exclude datelike types

rapidsai · Jul 18, 2024 · b21fdae · b21fdae
1 parent acd7cde
commit b21fdae
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/python/cudf/cudf/core/index.py b/python/cudf/cudf/core/index.py
@@ -355,12 +355,10 @@ def _data(self):
     @_performance_tracking
     def __contains__(self, item):
         hash(item)
-        if isinstance(item, bool) or not isinstance(
-            item, (np.floating, np.integer, int, float)
-        ):
+        if not isinstance(item, (np.floating, np.integer, int, float)):
             return False
-        elif isinstance(item, (np.timedelta64, np.datetime64)):
-            # Are instances of np.integer
+        elif isinstance(item, (np.timedelta64, np.datetime64, bool)):
+            # Cases that would pass the above check
             return False
         try:
             int_item = int(item)