Merge pull request #4010 from alejoe91/fix-get-neo-blocks-streams

Fix get_neo_* functions

Author: alejoe91

src/spikeinterface/extractors/neoextractors/__init__.py

@@ -86,3 +86,18 @@
     SpikeGLXEventExtractor: dict(wrapper_string="read_spikeglx_event", wrapper_class=read_spikeglx_event),
     MaxwellEventExtractor: dict(wrapper_string="read_maxwell_event", wrapper_class=read_maxwell_event),
 }
+
+
+# Utils dicts used for get_neo_extractor and get_neo_streams
+neo_recording_class_dict = {
+    rec_class.__name__.replace("Recording", "").replace("Extractor", "").lower(): rec_class
+    for rec_class in neo_recording_extractors_dict.keys()
+}
+neo_sorting_class_dict = {
+    sort_class.__name__.replace("Sorting", "").replace("Extractor", "").lower(): sort_class
+    for sort_class in neo_sorting_extractors_dict.keys()
+}
+neo_event_class_dict = {
+    event_class.__name__.replace("Event", "").replace("Extractor", "").lower(): event_class
+    for event_class in neo_event_extractors_dict.keys()
+}

src/spikeinterface/extractors/neoextractors/neo_utils.py

@@ -56,11 +56,10 @@ def get_neo_num_blocks(extractor_name, *args, **kwargs) -> int:
 
 
 def get_neo_extractor(extractor_name):
-    from spikeinterface.extractors.extractor_classes import recording_extractor_full_dict
+    from spikeinterface.extractors.neoextractors import neo_recording_class_dict
 
-    assert extractor_name in recording_extractor_full_dict, (
-        f"{extractor_name} not an extractor name:" f"\n{list(recording_extractor_full_dict.keys())}"
+    assert extractor_name in neo_recording_class_dict, (
+        f"{extractor_name} not an extractor name:" f"\n{list(neo_recording_class_dict.keys())}"
     )
-    neo_extractor = recording_extractor_full_dict[extractor_name]
-    assert issubclass(neo_extractor, NeoBaseRecordingExtractor), f"{extractor_name} is not a NEO recording extractor!"
+    neo_extractor = neo_recording_class_dict[extractor_name]
     return neo_extractor