Skip to content

Support int_range (Eager-only) #2916

@dangotbanned

Description

@dangotbanned

Eager Support

If you stare at (#2683) close enough - we've got most of it implemented for eager backends already.
We'd just need to expose start, step - where all we've used is stop/end:

row_index = pa.array(np.arange(frame.num_rows, dtype=np.int64))

if self._implementation.is_cudf():
import cupy as cp # ignore-banned-import # cuDF dependency.
data = cp.arange(size)
else:
import numpy as np # ignore-banned-import
data = np.arange(size)

pl.int_range(0, pl.len()).sort_by(order_by).alias(name), pl.all()

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions