Skip to content

Commit 911322d

Browse files
committed
Fix polars
1 parent f9fcbcd commit 911322d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

streamjoy/serializers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -259,8 +259,8 @@ def serialize_polars(
259259
groupby = kwargs.get("groupby")
260260

261261
if groupby:
262-
group_sizes = resources.groupby(groupby).agg(pl.len()).sort(by="count")
263-
total_frames = group_sizes.select(pl.col("count").max()).to_numpy()[0, 0]
262+
group_sizes = resources.groupby(groupby).agg(pl.len())
263+
total_frames = group_sizes.select(pl.col("len").max()).to_numpy()[0, 0]
264264
else:
265265
total_frames = len(resources)
266266

0 commit comments

Comments
 (0)