Skip to content

Commit 3642020

Browse files
authored
Merge pull request #4178 from MGAMZ/MGAMZ-Improve_job_kwarg_violation_hint
Improve assertion error message of SortingAnalyzer job kwargs
2 parents 24b7713 + 2132fa9 commit 3642020

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

src/spikeinterface/core/sortinganalyzer.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1634,11 +1634,17 @@ def compute(self, input, save=True, extension_params=None, verbose=False, **kwar
16341634
return self.compute_one_extension(extension_name=input, save=save, verbose=verbose, **kwargs)
16351635
elif isinstance(input, dict):
16361636
params_, job_kwargs = split_job_kwargs(kwargs)
1637-
assert len(params_) == 0, "Too many arguments for SortingAnalyzer.compute_several_extensions()"
1637+
assert len(params_) == 0, (
1638+
"Too many arguments for SortingAnalyzer.compute_several_extensions(), "
1639+
f"please remove the arguments {set(params_)} from the compute function."
1640+
)
16381641
self.compute_several_extensions(extensions=input, save=save, verbose=verbose, **job_kwargs)
16391642
elif isinstance(input, list):
16401643
params_, job_kwargs = split_job_kwargs(kwargs)
1641-
assert len(params_) == 0, "Too many arguments for SortingAnalyzer.compute_several_extensions()"
1644+
assert len(params_) == 0, (
1645+
"Too many arguments for SortingAnalyzer.compute_several_extensions(), "
1646+
f"please remove the arguments {set(params_)} from the compute function."
1647+
)
16421648
extensions = {k: {} for k in input}
16431649
if extension_params is not None:
16441650
for ext_name, ext_params in extension_params.items():

0 commit comments

Comments
 (0)