Merge pull request #1791 from NeuralEnsemble/black-formatting

Black formatting

Author: zm711

neo/rawio/blackrockrawio.py

@@ -80,7 +80,6 @@
 from neo.core import NeoReadWriteError
 
 
-
 class BlackrockRawIO(BaseRawIO):
     """
     Class for reading data in from a file set recorded by the Blackrock (Cerebus) recording system.
@@ -1205,14 +1204,14 @@ def _read_nsx_data_spec_v30_ptp(self, nsx_nb):
     def _read_nev_header(self, spec, filename):
         """
         Extract nev header information for any specification version.
-        
+
         Parameters
         ----------
         spec : str
             The specification version (e.g., "2.1", "2.2", "2.3", "3.0")
         filename : str
             The NEV filename to read from
-        
+
         Returns
         -------
         nev_basic_header : np.ndarray
@@ -1222,7 +1221,7 @@ def _read_nev_header(self, spec, filename):
         """
         # Note: This function only uses the passed parameters, not self attributes
         # This makes it easy to convert to @staticmethod later
-        
+
         # basic header (same for all versions)
         dt0 = [
             # Set to "NEURALEV"
@@ -1263,10 +1262,9 @@ def _read_nev_header(self, spec, filename):
 
         raw_ext_header = np.memmap(filename, offset=offset_dt0, dtype=dt1, shape=shape, mode="r")
 
-
         # Get extended header types for this spec
         header_types = NEV_EXT_HEADER_TYPES_BY_SPEC[spec]
-        
+
         # Parse extended headers by packet type
         # Strategy: view() entire array first, then mask for efficiency
         # Since all NEV extended header packets are fixed-width (32 bytes), temporarily
@@ -1283,7 +1281,7 @@ def _read_nev_header(self, spec, filename):
     def _read_nev_data(self, spec, filename):
         """
         Extract nev data for any specification version.
-        
+
         Parameters
         ----------
         spec : str
@@ -1333,8 +1331,8 @@ def _read_nev_data(self, spec, filename):
                 masks[data_type] = (min_val <= raw_data["packet_id"]) & (raw_data["packet_id"] <= max_val)
             else:
                 # Equality check
-                masks[data_type] = (raw_data["packet_id"] == packet_id_spec)
-            
+                masks[data_type] = raw_data["packet_id"] == packet_id_spec
+
             types[data_type] = data_types[data_type](packet_size_bytes)
 
         event_segment_ids = self._get_event_segment_ids(raw_data, masks, spec)
@@ -1351,14 +1349,13 @@ def _read_nev_data(self, spec, filename):
 
         return data
 
-
     def _get_reset_event_mask(self, raw_event_data, masks, spec):
         """
         Extract mask for reset comment events in 2.3 .nev file
         """
         if "Comments" not in masks:
             return np.zeros(len(raw_event_data), dtype=bool)
-            
+
         restart_mask = np.logical_and(
             masks["Comments"],
             raw_event_data["value"] == b"\x00\x00\x00\x00\x00\x00critical load restart",
@@ -1520,8 +1517,6 @@ def _match_nsx_and_nev_segment_ids(self, nsx_nb):
                 if len(ev_ids):
                     ev_ids[:] = np.vectorize(new_nev_segment_id_mapping.__getitem__)(ev_ids)
 
-
-
     def _nev_params(self, param_name):
         """
         Returns wanted nev parameter.

neo/rawio/neuralynxrawio/neuralynxrawio.py

@@ -64,7 +64,7 @@
 from neo.rawio.neuralynxrawio.nlxheader import NlxHeader
 
 # Named tuple for stream identification keys
-StreamKey = namedtuple('StreamKey', ['sampling_rate', 'input_range', 'filter_params'])
+StreamKey = namedtuple("StreamKey", ["sampling_rate", "input_range", "filter_params"])
 
 
 class NeuralynxRawIO(BaseRawIO):
@@ -456,7 +456,9 @@ def _parse_header(self):
                 # Format stream name using namedtuple fields
                 dsp_filter_id = seen_filters[stream_key.filter_params]
                 voltage_mv = int(stream_key.input_range / 1000) if stream_key.input_range is not None else 0
-                stream_name = f"stream{stream_id}_{int(stream_key.sampling_rate)}Hz_{voltage_mv}mVRange_DSPFilter{dsp_filter_id}"
+                stream_name = (
+                    f"stream{stream_id}_{int(stream_key.sampling_rate)}Hz_{voltage_mv}mVRange_DSPFilter{dsp_filter_id}"
+                )
 
                 stream_names.append(stream_name)
                 stream_ids.append(str(stream_id))

neo/rawio/neuralynxrawio/nlxheader.py

@@ -367,31 +367,31 @@ def _normalize_types(self):
 
         # Convert boolean strings to actual booleans
         bool_keys = [
-            'DSPLowCutFilterEnabled',
-            'DSPHighCutFilterEnabled',
-            'DspDelayCompensation',
+            "DSPLowCutFilterEnabled",
+            "DSPHighCutFilterEnabled",
+            "DspDelayCompensation",
         ]
 
         for key in bool_keys:
             if key in self and isinstance(self[key], str):
-                if self[key] in ('True', 'Enabled'):
+                if self[key] in ("True", "Enabled"):
                     self[key] = True
-                elif self[key] in ('False', 'Disabled'):
+                elif self[key] in ("False", "Disabled"):
                     self[key] = False
 
         # Convert numeric strings to numbers
         numeric_keys = [
-            'DspLowCutFrequency',
-            'DspHighCutFrequency',
-            'DspLowCutNumTaps',
-            'DspHighCutNumTaps',
+            "DspLowCutFrequency",
+            "DspHighCutFrequency",
+            "DspLowCutNumTaps",
+            "DspHighCutNumTaps",
         ]
 
         for key in numeric_keys:
             if key in self and isinstance(self[key], str):
                 try:
                     # Try int first
-                    if '.' not in self[key]:
+                    if "." not in self[key]:
                         self[key] = int(self[key])
                     else:
                         self[key] = float(self[key])
@@ -400,7 +400,7 @@ def _normalize_types(self):
                     pass
 
         # Handle DspFilterDelay_µs (could be string or already converted)
-        delay_key = 'DspFilterDelay_µs'
+        delay_key = "DspFilterDelay_µs"
         if delay_key in self and isinstance(self[delay_key], str):
             try:
                 self[delay_key] = int(self[delay_key])
@@ -410,10 +410,10 @@ def _normalize_types(self):
         # Extract single-channel InputRange from list
         # For multi-channel files, keep as list
         # For single-channel files, extract the single value
-        if 'InputRange' in self and isinstance(self['InputRange'], list):
-            if len(self['InputRange']) == 1:
+        if "InputRange" in self and isinstance(self["InputRange"], list):
+            if len(self["InputRange"]) == 1:
                 # Single channel file: extract the value
-                self['InputRange'] = self['InputRange'][0]
+                self["InputRange"] = self["InputRange"][0]
             # else: multi-channel, keep as list
 
     def type_of_recording(self):