Update away from deprecated polars function names (#5618)

microsoft · Sep 5, 2023 · f6e0e34 · f6e0e34
1 parent 4d4a1a2
commit f6e0e34
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/infra/basicperf.py b/tests/infra/basicperf.py
@@ -418,8 +418,8 @@ def table():
                         overall = rcvd.join(overall, on="messageID")
                         overall = overall.with_columns(
                             client=pl.lit(remote_client.name),
-                            requestSize=pl.col("request").apply(len),
-                            responseSize=pl.col("rawResponse").apply(len),
+                            requestSize=pl.col("request").map_elements(len),
+                            responseSize=pl.col("rawResponse").map_elements(len),
                         )
                         # 50x are expected when we stop the primary, 500 when we drop the session
                         # to maintain consistency, and 504 when we try to write to the future primary
@@ -520,7 +520,7 @@ def table():
                             (pl.col("sendTime").alias("second") - start_send) / 1000000
                         ).cast(pl.Int64)
                     )
-                    .groupby("second")
+                    .group_by("second")
                     .count()
                     .rename({"count": "sent"})
                 )
@@ -531,7 +531,7 @@ def table():
                             / 1000000
                         ).cast(pl.Int64)
                     )
-                    .groupby("second")
+                    .group_by("second")
                     .count()
                     .rename({"count": "rcvd"})
                 )