Skip to content

Commit 4ce3e40

Browse files
committed
Call unique() and value_count() on Series and not list
1 parent ce60fac commit 4ce3e40

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/estimagic/parameters/parameter_groups.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,10 @@ def get_params_groups_and_short_names(params, free_mask, max_group_size=8):
3535
names.append(name)
3636

3737
# if every parameter has its own group, they should all actually be in one group
38-
if len(pd.unique(groups)) == len(groups):
38+
if len(pd.Series(groups).unique()) == len(groups):
3939
groups = ["Parameters"] * len(groups)
4040

41-
counts = pd.value_counts(groups)
41+
counts = pd.Series(groups).value_counts()
4242
to_be_split = counts[counts > max_group_size]
4343
for group_name, n_occurrences in to_be_split.items():
4444
split_group_names = _split_long_group(

0 commit comments

Comments
 (0)