@@ -304,15 +304,14 @@ def _plc_windows(self) -> WindowTypePair:
304304 orderby_obj = as_column (range (len (self .obj )))
305305 if self ._group_keys is not None :
306306 group_cols : list [plc .Column ] = [
307- col .to_pylibcudf (mode = "read" )
308- for col in self ._group_keys ._columns
307+ col .plc_column for col in self ._group_keys ._columns
309308 ]
310309 else :
311310 group_cols = []
312311 group_keys = plc .Table (group_cols )
313312 return plc .rolling .make_range_windows (
314313 group_keys ,
315- orderby_obj .to_pylibcudf ( mode = "read" ) ,
314+ orderby_obj .plc_column ,
316315 plc .types .Order .ASCENDING ,
317316 plc .types .NullOrder .BEFORE ,
318317 plc .rolling .BoundedOpen (plc .Scalar .from_py (pre )),
@@ -337,8 +336,8 @@ def _plc_windows(self) -> WindowTypePair:
337336 SIZE_TYPE_DTYPE
338337 )
339338 return (
340- preceding_window .to_pylibcudf ( mode = "read" ) ,
341- following_window .to_pylibcudf ( mode = "read" ) ,
339+ preceding_window .plc_column ,
340+ following_window .plc_column ,
342341 )
343342 else :
344343 raise ValueError (
@@ -359,7 +358,7 @@ def _apply_agg_column(
359358 with acquire_spill_lock ():
360359 return ColumnBase .from_pylibcudf (
361360 plc .rolling .rolling_window (
362- source_column .to_pylibcudf ( mode = "read" ) ,
361+ source_column .plc_column ,
363362 pre ,
364363 fwd ,
365364 self .min_periods or 1 ,
0 commit comments