Add in cast that was lost in the scramble.

rapidsai · Aug 12, 2021 · 939b455 · 939b455
1 parent 745684f
commit 939b455
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/cudf/cudf/core/frame.py b/python/cudf/cudf/core/frame.py
@@ -4316,7 +4316,9 @@ def cumsum(self, axis=None, skipna=True, *args, **kwargs):
         2   6  24
         3  10  34
         """
-        return self._scan("sum", axis=axis, skipna=skipna, *args, **kwargs)
+        return self._scan(
+            "sum", axis=axis, skipna=skipna, cast_to_int=True, *args, **kwargs
+        )
 
     def cumprod(self, axis=None, skipna=True, *args, **kwargs):
         """
@@ -4346,7 +4348,9 @@ def cumprod(self, axis=None, skipna=True, *args, **kwargs):
         2   6   504
         3  24  5040
         """
-        return self._scan("prod", axis=axis, skipna=skipna, *args, **kwargs)
+        return self._scan(
+            "prod", axis=axis, skipna=skipna, cast_to_int=True, *args, **kwargs
+        )
 
 
 class SingleColumnFrame(Frame):