Skip to content

Commit

Permalink
Merge pull request #3430 from alejoe91/fix-get-data-compatibility
Browse files Browse the repository at this point in the history
Set `run_info` to `None` for `load_waveforms`
  • Loading branch information
samuelgarcia authored Sep 23, 2024
2 parents 0e72755 + dc3a026 commit a874f2a
Showing 1 changed file with 4 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -536,6 +536,7 @@ def _read_old_waveforms_extractor_binary(folder, sorting):
ext = ComputeRandomSpikes(sorting_analyzer)
ext.params = dict()
ext.data = dict(random_spikes_indices=random_spikes_indices)
ext.run_info = None
sorting_analyzer.extensions["random_spikes"] = ext

ext = ComputeWaveforms(sorting_analyzer)
Expand All @@ -545,6 +546,7 @@ def _read_old_waveforms_extractor_binary(folder, sorting):
dtype=params["dtype"],
)
ext.data["waveforms"] = waveforms
ext.run_info = None
sorting_analyzer.extensions["waveforms"] = ext

# templates saved dense
Expand All @@ -559,6 +561,7 @@ def _read_old_waveforms_extractor_binary(folder, sorting):
ext.params = dict(ms_before=params["ms_before"], ms_after=params["ms_after"], operators=list(templates.keys()))
for mode, arr in templates.items():
ext.data[mode] = arr
ext.run_info = None
sorting_analyzer.extensions["templates"] = ext

for old_name, new_name in old_extension_to_new_class_map.items():
Expand Down Expand Up @@ -631,6 +634,7 @@ def _read_old_waveforms_extractor_binary(folder, sorting):
ext.set_params(**updated_params, save=False)
if ext.need_backward_compatibility_on_load:
ext._handle_backward_compatibility_on_load()
ext.run_info = None

sorting_analyzer.extensions[new_name] = ext

Expand Down

0 comments on commit a874f2a

Please sign in to comment.