improve logic

Author: alejoe91

src/probeinterface/neuropixels_tools.py

@@ -1004,15 +1004,13 @@ def read_openephys(
         for stream_field in stream_fields:
             stream = stream_field.attrib["name"]
             # exclude ADC streams
-            if "ADC" not in stream:
-                streams.append(stream)
-            # find probe name (exclude AP-LFP from stream name)
-            if "AP" in stream or "LFP" in stream:
-                probe_name_ = stream.split("-")[0]
-            else:
-                probe_name_ = stream
-            if probe_name_ not in probe_names_used:
-                probe_names_used.append(probe_name_)
+            if "ADC" in stream:
+                continue
+            streams.append(stream)
+            # find probe name (exclude "-AP"/"-LFP" from stream name)
+            stream = stream.replace("-AP", "").replace("-LFP", "")
+            if stream not in probe_names_used:
+                probe_names_used.append(stream)
     else:
         has_streams = False
         probe_names_used = None