Dtypes aren't callable

rapidsai · Mar 22, 2023 · 7998d90 · 7998d90
1 parent adb7a0b
commit 7998d90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf/core/groupby/groupby.py b/python/cudf/cudf/core/groupby/groupby.py
@@ -945,7 +945,7 @@ def sample(
         size_per_group = np.diff(group_offsets)
         if n is not None:
             samples_per_group = np.broadcast_to(
-                size_type_dtype(n), size_per_group.shape
+                size_type_dtype.type(n), size_per_group.shape
             )
             if not replace and (minsize := size_per_group.min()) < n:
                 raise ValueError(