diff --git a/spec/API_specification/dataframe_api/dataframe_object.py b/spec/API_specification/dataframe_api/dataframe_object.py index 45269423..c48464b7 100644 --- a/spec/API_specification/dataframe_api/dataframe_object.py +++ b/spec/API_specification/dataframe_api/dataframe_object.py @@ -88,24 +88,18 @@ def get_columns_by_name(self, names: Sequence[str], /) -> DataFrame: """ ... - def get_rows(self, indices: Sequence[int]) -> DataFrame: + def get_rows(self, indices: "Column[int]") -> DataFrame: """ Select a subset of rows, similar to `ndarray.take`. Parameters ---------- - indices : Sequence[int] + indices : Column[int] Positions of rows to select. Returns ------- DataFrame - - Notes - ----- - Some discussion participants prefer a stricter type Column[int] for - indices in order to make it easier to implement in a performant manner - on GPUs. """ ...