use np.generic

rjzamora · Jan 9, 2025 · 20ab966 · 20ab966
1 parent e75a9bf
commit 20ab966
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dask/dataframe/dask_expr/_collection.py b/dask/dataframe/dask_expr/_collection.py
@@ -416,7 +416,7 @@ def __getitem__(self, other):
             other = list(other)
         elif isinstance(other, list):
             other = other.copy()
-        elif is_scalar(other) and hasattr(other, "item"):
+        elif isinstance(other, np.generic):
             other = other.item()
         return new_collection(self.expr.__getitem__(other))
 

diff --git a/dask/dataframe/dask_expr/_indexing.py b/dask/dataframe/dask_expr/_indexing.py
@@ -78,7 +78,7 @@ def __getitem__(self, key):
             iindexer = key
             cindexer = None
 
-        if is_scalar(cindexer) and hasattr(cindexer, "item"):
+        if isinstance(cindexer, np.generic):
             cindexer = cindexer.item()
 
         return self._loc(iindexer, cindexer)