From 9de041c86f394ac016f783787af435d618989ff2 Mon Sep 17 00:00:00 2001 From: Marco Gorelli <33491632+MarcoGorelli@users.noreply.github.com> Date: Fri, 21 Jun 2024 13:13:44 +0100 Subject: [PATCH] enforce keyword-only --- py-polars/polars/dataframe/frame.py | 3 ++- py-polars/polars/lazyframe/frame.py | 2 +- py-polars/tests/unit/operations/test_filter.py | 2 +- py-polars/tests/unit/operations/test_unpivot.py | 4 ++-- 4 files changed, 6 insertions(+), 5 deletions(-) diff --git a/py-polars/polars/dataframe/frame.py b/py-polars/polars/dataframe/frame.py index 8d06119f3a74..f18fd1ab23f0 100644 --- a/py-polars/polars/dataframe/frame.py +++ b/py-polars/polars/dataframe/frame.py @@ -7589,11 +7589,11 @@ def explode( @deprecate_renamed_parameter("columns", "on", version="1.0.0") def pivot( self, + *, on: ColumnNameOrSelector | Sequence[ColumnNameOrSelector] | None, index: ColumnNameOrSelector | Sequence[ColumnNameOrSelector] | None, values: ColumnNameOrSelector | Sequence[ColumnNameOrSelector] | None, aggregate_function: PivotAgg | Expr | None = None, - *, maintain_order: bool = True, sort_columns: bool = False, separator: str = "_", @@ -7805,6 +7805,7 @@ def pivot( def unpivot( self, + *, on: ColumnNameOrSelector | Sequence[ColumnNameOrSelector] | None = None, index: ColumnNameOrSelector | Sequence[ColumnNameOrSelector] | None = None, variable_name: str | None = None, diff --git a/py-polars/polars/lazyframe/frame.py b/py-polars/polars/lazyframe/frame.py index b0bcd0d50dcc..5e0728a937cd 100644 --- a/py-polars/polars/lazyframe/frame.py +++ b/py-polars/polars/lazyframe/frame.py @@ -5705,11 +5705,11 @@ def drop_nulls( def unpivot( self, + *, on: ColumnNameOrSelector | Sequence[ColumnNameOrSelector] | None = None, index: ColumnNameOrSelector | Sequence[ColumnNameOrSelector] | None = None, variable_name: str | None = None, value_name: str | None = None, - *, streamable: bool = True, ) -> Self: """ diff --git a/py-polars/tests/unit/operations/test_filter.py b/py-polars/tests/unit/operations/test_filter.py index 85db19b228aa..d1c0de20a04b 100644 --- a/py-polars/tests/unit/operations/test_filter.py +++ b/py-polars/tests/unit/operations/test_filter.py @@ -40,7 +40,7 @@ def test_unpivot_values_predicate_pushdown() -> None: ).lazy() assert ( - lf.unpivot("id", ["asset_key_1", "asset_key_2", "asset_key_3"]) + lf.unpivot(index="id", on=["asset_key_1", "asset_key_2", "asset_key_3"]) .filter(pl.col("value") == pl.lit("123")) .collect() ).to_dict(as_series=False) == { diff --git a/py-polars/tests/unit/operations/test_unpivot.py b/py-polars/tests/unit/operations/test_unpivot.py index 9dc758d1eb98..a4155da56874 100644 --- a/py-polars/tests/unit/operations/test_unpivot.py +++ b/py-polars/tests/unit/operations/test_unpivot.py @@ -76,10 +76,10 @@ def test_unpivot_projection_pd_7747() -> None: # https://github.com/pola-rs/polars/issues/10075 -def test_unpivot_no_value_vars() -> None: +def test_unpivot_no_on() -> None: lf = pl.LazyFrame({"a": [1, 2, 3]}) - result = lf.unpivot("a") + result = lf.unpivot(index="a") expected = pl.LazyFrame( schema={"a": pl.Int64, "variable": pl.String, "value": pl.Null}