cudf-polars-cu13 25.10.0__py3-none-any.whl → 26.2.0__py3-none-any.whl

cudf_polars/dsl/expressions/selection.py

@@ -35,22 +35,23 @@ class Gather(Expr):
         values, indices = (
             child.evaluate(df, context=context) for child in self.children
         )
-        lo, hi = plc.reduce.minmax(indices.obj)
-        lo = lo.to_py()
-        hi = hi.to_py()
-        n = df.num_rows
-        if hi >= n or lo < -n:
+        n = values.size
+        lo, hi = plc.reduce.minmax(indices.obj, stream=df.stream)
+        if hi.to_py(stream=df.stream) >= n or lo.to_py(stream=df.stream) < -n:  # type: ignore[operator]
             raise ValueError("gather indices are out of bounds")
         if indices.null_count:
             bounds_policy = plc.copying.OutOfBoundsPolicy.NULLIFY
             obj = plc.replace.replace_nulls(
                 indices.obj,
-                plc.Scalar.from_py(n, dtype=indices.obj.type()),
+                plc.Scalar.from_py(n, dtype=indices.obj.type(), stream=df.stream),
+                stream=df.stream,
             )
         else:
             bounds_policy = plc.copying.OutOfBoundsPolicy.DONT_CHECK
             obj = indices.obj
-        table = plc.copying.gather(plc.Table([values.obj]), obj, bounds_policy)
+        table = plc.copying.gather(
+            plc.Table([values.obj]), obj, bounds_policy, stream=df.stream
+        )
         return Column(table.columns()[0], dtype=self.dtype)
 
 
@@ -69,6 +70,6 @@ class Filter(Expr):
         """Evaluate this expression given a dataframe for context."""
         values, mask = (child.evaluate(df, context=context) for child in self.children)
         table = plc.stream_compaction.apply_boolean_mask(
-            plc.Table([values.obj]), mask.obj
+            plc.Table([values.obj]), mask.obj, stream=df.stream
         )
         return Column(table.columns()[0], dtype=self.dtype).sorted_like(values)

cudf_polars/dsl/expressions/slicing.py

@@ -43,4 +43,4 @@ class Slice(Expr):
         """Evaluate this expression given a dataframe for context."""
         (child,) = self.children
         column = child.evaluate(df, context=context)
-        return column.slice((self.offset, self.length))
+        return column.slice((self.offset, self.length), stream=df.stream)