From f066519f7e34755e6b2e37c3f70f74537ae46f78 Mon Sep 17 00:00:00 2001 From: Takanori Hayashi Date: Fri, 6 Dec 2024 18:28:05 +0900 Subject: [PATCH] Stop passing ddof to pl.corr in Python unit test --- py-polars/tests/unit/lazyframe/test_lazyframe.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/py-polars/tests/unit/lazyframe/test_lazyframe.py b/py-polars/tests/unit/lazyframe/test_lazyframe.py index d738f9111c2e..4aafc9d523ee 100644 --- a/py-polars/tests/unit/lazyframe/test_lazyframe.py +++ b/py-polars/tests/unit/lazyframe/test_lazyframe.py @@ -946,7 +946,7 @@ def test_spearman_corr_ties() -> None: df = pl.DataFrame({"a": [1, 1, 1, 2, 3, 7, 4], "b": [4, 3, 2, 2, 4, 3, 1]}) result = df.select( - pl.corr("a", "b", method="spearman", ddof=0).alias("a1"), + pl.corr("a", "b", method="spearman").alias("a1"), pl.corr(pl.col("a").rank("min"), pl.col("b").rank("min")).alias("a2"), pl.corr(pl.col("a").rank(), pl.col("b").rank()).alias("a3"), ) @@ -972,7 +972,9 @@ def test_pearson_corr() -> None: out = ( ldf.group_by("era", maintain_order=True).agg( pl.corr( - pl.col("prediction"), pl.col("target"), method="pearson", ddof=0 + pl.col("prediction"), + pl.col("target"), + method="pearson", ).alias("c"), ) ).collect()["c"] @@ -981,7 +983,7 @@ def test_pearson_corr() -> None: # we can also pass in column names directly out = ( ldf.group_by("era", maintain_order=True).agg( - pl.corr("prediction", "target", method="pearson", ddof=0).alias("c"), + pl.corr("prediction", "target", method="pearson").alias("c"), ) ).collect()["c"] assert out.to_list() == pytest.approx([0.6546536707079772, -5.477514993831792e-1])