From b8bcf1c46232c5836432fad924ad638a367231c3 Mon Sep 17 00:00:00 2001 From: Doug Branton Date: Fri, 26 Jul 2024 14:27:14 -0700 Subject: [PATCH] formatting --- src/nested_pandas/utils/utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/nested_pandas/utils/utils.py b/src/nested_pandas/utils/utils.py index ea17ce1..e337dcf 100644 --- a/src/nested_pandas/utils/utils.py +++ b/src/nested_pandas/utils/utils.py @@ -29,9 +29,7 @@ def count_nested(df, nested, by=None, join=True) -> NestedFrame: if by is None: field_to_len = df[nested].nest.fields[0] - counts = ( - df[nested].nest.to_lists().apply(lambda x: len(x[field_to_len]), axis=1) - ) + counts = df[nested].nest.to_lists().apply(lambda x: len(x[field_to_len]), axis=1) counts.name = f"n_{nested}" # update name directly (rename causes issues downstream) else: # this may be able to be sped up using tolists() as well