handles case when periods > len(df)

rapidsai · Feb 17, 2022 · c264ea4 · c264ea4
1 parent a4f1428
commit c264ea4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/python/cudf/cudf/core/column/column.py b/python/cudf/cudf/core/column/column.py
@@ -335,6 +335,12 @@ def _fill(
         return self
 
     def shift(self, offset: int, fill_value: ScalarLike) -> ColumnBase:
+        # link to the libcudf ticket you will create
+        if abs(offset) > len(self):
+            if fill_value is None:
+                return column_empty_like(self, masked=True)
+            else:
+                return full(len(self), fill_value, dtype=self.dtype)
         return libcudf.copying.shift(self, offset, fill_value)
 
     @property