Skip to content

Commit

Permalink
Fix old sparsity representation
Browse files Browse the repository at this point in the history
  • Loading branch information
alejoe91 committed May 21, 2024
1 parent 311440a commit 21b726b
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions src/spikeinterface/core/waveform_extractor.py
Original file line number Diff line number Diff line change
Expand Up @@ -1928,11 +1928,17 @@ def load(cls, folder, waveform_extractor):
params = cls.load_params_from_folder(folder)

if "sparsity" in params and params["sparsity"] is not None:
sparsity_params = params["sparsity"]
# handle old sparsity version
if "unit_ids" not in params["sparsity"]:
sparsity_params = {}
sparsity_params["unit_ids"] = waveform_extractor.unit_ids
sparsity_params["channel_ids"] = waveform_extractor.channel_ids
sparsity_params["unit_id_to_channel_ids"] = params["sparsity"]
else:
sparsity_params = params["sparsity"]
params["sparsity"] = ChannelSparsity.from_dict(params["sparsity"])

# if waveform_extractor is None:
# waveform_extractor = WaveformExtractor.load(folder)

# make instance with params
ext = cls(waveform_extractor)
ext._params = params
Expand Down

0 comments on commit 21b726b

Please sign in to comment.