Skip to content

Commit

Permalink
Merge pull request #3220 from zm711/docstring-analzyer
Browse files Browse the repository at this point in the history
Analyzer docstring cleanup
  • Loading branch information
samuelgarcia authored Jul 18, 2024
2 parents 5e222d3 + 446684d commit 60d7196
Showing 1 changed file with 27 additions and 24 deletions.
51 changes: 27 additions & 24 deletions src/spikeinterface/core/sortinganalyzer.py
Original file line number Diff line number Diff line change
Expand Up @@ -815,10 +815,10 @@ def save_as(self, format="memory", folder=None) -> "SortingAnalyzer":
Parameters
----------
folder : str or Path
The output waveform folder
format : "binary_folder" | "zarr", default: "binary_folder"
The backend to use for saving the waveforms
folder : str | Path | None, default: None
The output folder if `format` is "zarr" or "binary_folder"
format : "memory" | "binary_folder" | "zarr", default: "memory"
The new backend format to use
"""
return self._save_or_select_or_merge(format=format, folder=folder)

Expand All @@ -835,8 +835,9 @@ def select_units(self, unit_ids, format="memory", folder=None) -> "SortingAnalyz
The unit ids to keep in the new SortingAnalyzer object
format : "memory" | "binary_folder" | "zarr" , default: "memory"
The format of the returned SortingAnalyzer.
folder : Path or None
The new folder where selected waveforms are copied.
folder : Path | None, deafult: None
The new folder where the analyzer with selected units is copied if `format` is
"binary_folder" or "zarr"
Returns
-------
Expand All @@ -859,8 +860,9 @@ def remove_units(self, remove_unit_ids, format="memory", folder=None) -> "Sortin
The unit ids to remove in the new SortingAnalyzer object.
format : "memory" | "binary_folder" | "zarr" , default: "memory"
The format of the returned SortingAnalyzer.
folder : Path or None
The new folder where selected waveforms are copied.
folder : Path or None, default: None
The new folder where the analyzer without removed units is copied if `format`
is "binary_folder" or "zarr"
Returns
-------
Expand All @@ -886,42 +888,43 @@ def merge_units(
) -> "SortingAnalyzer":
"""
This method is equivalent to `save_as()`but with a list of merges that have to be achieved.
Merges units by creating a new sorting analyzer object in a new folder with appropriate merges
Merges units by creating a new SortingAnalyzer object with the appropriate merges
Extensions are also updated to display the merged unit ids.
Extensions are also updated to display the merged `unit_ids`.
Parameters
----------
merge_unit_groups : list/tuple of lists/tuples
A list of lists for every merge group. Each element needs to have at least two elements (two units to merge),
but it can also have more (merge multiple units at once).
new_unit_ids : None or list
new_unit_ids : None | list, default: None
A new unit_ids for merged units. If given, it needs to have the same length as `merge_unit_groups`. If None,
merged units will have the first unit_id of every lists of merges
censor_ms : None or float
When merging units, any spikes violating this refractory period will be discarded. Default is None
merging_mode : "soft" can be in ["soft", "hard"]
How merges are performed. In the "soft" mode, merges will be approximated, with no reloading of the
waveforms. This will lead to approximations. If "hard", recomputations are accuratly performed,
censor_ms : None | float, default: None
When merging units, any spikes violating this refractory period will be discarded. If None all units are kept
merging_mode : ["soft", "hard"], default: "soft"
How merges are performed. If the `merge_mode` is "soft" , merges will be approximated, with no reloading of the
waveforms. This will lead to approximations. If `merge_mode` is "hard", recomputations are accurately performed,
reloading waveforms if needed
sparsity_overlap : float, default 0.75
The percentage of overlap that units should share in order to accept merges. If this criteria is not
achieved, soft merging will not be possible
achieved, soft merging will not be possible and an error will be raised
new_id_strategy : "append" | "take_first", default: "append"
The strategy that should be used, if `new_unit_ids` is None, to create new unit_ids.
* "append" : new_units_ids will be added at the end of max(sorging.unit_ids)
* "append" : new_units_ids will be added at the end of max(sorting.unit_ids)
* "take_first" : new_unit_ids will be the first unit_id of every list of merges
folder : Path or None
The new folder where selected waveforms are copied
format : "auto" | "binary_folder" | "zarr"
The format of the folder.
verbose:
folder : Path | None, default: None
The new folder where the analyzer with merged units is copied for `format` "binary_folder" or "zarr"
format : "memory" | "binary_folder" | "zarr", default: "memory"
The format of SortingAnalyzer
verbose : bool, default: False
Whether to display calculations (such as sparsity estimation)
Returns
-------
analyzer : SortingAnalyzer
The newly create sorting_analyzer with the selected units
The newly create `SortingAnalyzer` with the selected units
"""

assert merging_mode in ["soft", "hard"], "Merging mode should be either soft or hard"
Expand Down

0 comments on commit 60d7196

Please sign in to comment.