diff --git a/spikeinterface/extractors/neoextractors/ced.py b/spikeinterface/extractors/neoextractors/ced.py index 3b6483c224..b2870acab2 100644 --- a/spikeinterface/extractors/neoextractors/ced.py +++ b/spikeinterface/extractors/neoextractors/ced.py @@ -27,7 +27,7 @@ def __init__(self, file_path, stream_id=None, all_annotations=False): neo_kwargs = {'filename': str(file_path)} NeoBaseRecordingExtractor.__init__(self, stream_id=stream_id, all_annotations=all_annotations, **neo_kwargs) self._kwargs.update(dict(file_path=str(file_path))) - self.extra_requirements.append('sonpy') + self.extra_requirements.append('neo[ced]') diff --git a/spikeinterface/extractors/neoextractors/edf.py b/spikeinterface/extractors/neoextractors/edf.py index 6407be5d90..0696762cc3 100644 --- a/spikeinterface/extractors/neoextractors/edf.py +++ b/spikeinterface/extractors/neoextractors/edf.py @@ -26,7 +26,7 @@ def __init__(self, file_path, stream_id=None, all_annotations=False): neo_kwargs = {'filename': str(file_path)} NeoBaseRecordingExtractor.__init__(self, stream_id=stream_id, all_annotations=all_annotations, **neo_kwargs) self._kwargs.update({'file_path': str(file_path)}) - self.extra_requirements.append('pyedflib') + self.extra_requirements.append('neo[edf]') -read_edf = define_function_from_class(source_class=EDFRecordingExtractor, name="read_edf") \ No newline at end of file +read_edf = define_function_from_class(source_class=EDFRecordingExtractor, name="read_edf") diff --git a/spikeinterface/extractors/neoextractors/nix.py b/spikeinterface/extractors/neoextractors/nix.py index 358b3943c2..b88ef731fb 100644 --- a/spikeinterface/extractors/neoextractors/nix.py +++ b/spikeinterface/extractors/neoextractors/nix.py @@ -30,7 +30,7 @@ def __init__(self, file_path, stream_id=None, all_annotations=False): neo_kwargs = {'filename': str(file_path)} NeoBaseRecordingExtractor.__init__(self, stream_id=stream_id, all_annotations=all_annotations, **neo_kwargs) self._kwargs.update(dict(file_path=str(file_path), stream_id=stream_id)) - self.extra_requirements.append('nixio') + self.extra_requirements.append('neo[nixio]') read_nix = define_function_from_class(source_class=NixRecordingExtractor, name="read_nix") diff --git a/spikeinterface/extractors/neoextractors/openephys.py b/spikeinterface/extractors/neoextractors/openephys.py index dfcdd4c80b..349db4b3e6 100644 --- a/spikeinterface/extractors/neoextractors/openephys.py +++ b/spikeinterface/extractors/neoextractors/openephys.py @@ -93,7 +93,6 @@ def __init__(self, folder_path, stream_id=None, all_annotations=False): sample_shifts = get_neuropixels_sample_shifts(self.get_num_channels(), num_channels_per_adc) self.set_property("inter_sample_shift", sample_shifts) - self._kwargs .update(dict(folder_path=str(folder_path)))