update to all float types

rapidsai · Mar 1, 2021 · 30ca48a · 30ca48a
1 parent f203cb2
commit 30ca48a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf/core/frame.py b/python/cudf/cudf/core/frame.py
@@ -1195,7 +1195,7 @@ def scatter_by_map(
             map_index = as_column(map_index)
 
         # Convert float to integer
-        if map_index.dtype == float:
+        if map_index.dtype.kind == "f":
             map_index = map_index.astype(np.int32)
 
         # Convert string or categorical to integer