Run black formatting

Author: glopesdev

harp/io.py

@@ -73,9 +73,7 @@ def read(
     """
     data = np.fromfile(file, dtype=np.uint8)
     if len(data) == 0:
-        return pd.DataFrame(
-            columns=columns, index=pd.Index([], dtype=np.float64, name="Time")
-        )
+        return pd.DataFrame(columns=columns, index=pd.Index([], dtype=np.float64, name="Time"))
 
     if address is not None and address != data[2]:
         raise ValueError(f"expected address {address} but got {data[2]}")
@@ -86,13 +84,9 @@ def read(
     payloadtype = data[4]
     payloadoffset = 5
     if payloadtype & 0x10 != 0:
-        seconds = np.ndarray(
-            nrows, dtype=np.uint32, buffer=data, offset=payloadoffset, strides=stride
-        )
+        seconds = np.ndarray(nrows, dtype=np.uint32, buffer=data, offset=payloadoffset, strides=stride)
         payloadoffset += 4
-        micros = np.ndarray(
-            nrows, dtype=np.uint16, buffer=data, offset=payloadoffset, strides=stride
-        )
+        micros = np.ndarray(nrows, dtype=np.uint16, buffer=data, offset=payloadoffset, strides=stride)
         payloadoffset += 2
         time = micros * _SECONDS_PER_TICK + seconds
         payloadtype = payloadtype & ~0x10
@@ -121,9 +115,7 @@ def read(
 
     result = pd.DataFrame(payload, index=index, columns=columns)
     if keep_type:
-        msgtype = np.ndarray(
-            nrows, dtype=np.uint8, buffer=data, offset=0, strides=stride
-        )
+        msgtype = np.ndarray(nrows, dtype=np.uint8, buffer=data, offset=0, strides=stride)
         msgtype = pd.Categorical.from_codes(msgtype, categories=_messagetypes)  # type: ignore
         result[MessageType.__name__] = msgtype
     return result

harp/model.py

@@ -136,16 +136,12 @@ class Visibility(Enum):
 class Register(BaseModel):
     address: Annotated[
         int,
-        Field(
-            le=255, description="Specifies the unique 8-bit address of the register."
-        ),
+        Field(le=255, description="Specifies the unique 8-bit address of the register."),
     ]
     type: Annotated[PayloadType, BeforeValidator(lambda v: PayloadType[v])]
     length: Annotated[
         Optional[int],
-        Field(
-            ge=1, default=1, description="Specifies the length of the register payload."
-        ),
+        Field(ge=1, default=1, description="Specifies the length of the register payload."),
     ]
     access: Union[Access, List[Access]] = Field(
         ..., description="Specifies the expected use of the register."
@@ -191,12 +187,6 @@ class Registers(BaseModel):
 
 class Model(Registers):
     device: str = Field(..., description="Specifies the name of the device.")
-    whoAmI: int = Field(
-        ..., description="Specifies the unique identifier for this device type."
-    )
-    firmwareVersion: str = Field(
-        ..., description="Specifies the semantic version of the device firmware."
-    )
-    hardwareTargets: str = Field(
-        ..., description="Specifies the semantic version of the device hardware."
-    )
+    whoAmI: int = Field(..., description="Specifies the unique identifier for this device type.")
+    firmwareVersion: str = Field(..., description="Specifies the semantic version of the device firmware.")
+    hardwareTargets: str = Field(..., description="Specifies the semantic version of the device hardware.")

harp/reader.py

@@ -28,8 +28,7 @@ def __call__(
         file: Optional[Union[str, bytes, PathLike[Any], BinaryIO]] = None,
         epoch: Optional[datetime] = None,
         keep_type: bool = False,
-    ) -> DataFrame:
-        ...
+    ) -> DataFrame: ...
 
 
 class RegisterReader:
@@ -50,9 +49,7 @@ class RegisterMap(UserDict[str, RegisterReader]):
 
     def __init__(self, registers: Mapping[str, RegisterReader]) -> None:
         super().__init__(registers)
-        self._address_map = {
-            value.register.address: value for value in registers.values()
-        }
+        self._address_map = {value.register.address: value for value in registers.values()}
 
     def __getitem__(self, __key: Union[str, int]) -> RegisterReader:
         if isinstance(__key, int):
@@ -266,9 +263,7 @@ def create_reader(
         base_path = path / device.device if is_dir else path.parent / device.device
 
     reg_readers = {
-        name: _create_register_parser(
-            device, name, _ReaderParams(base_path, epoch, keep_type)
-        )
+        name: _create_register_parser(device, name, _ReaderParams(base_path, epoch, keep_type))
         for name in device.registers.keys()
     }
     return DeviceReader(device, reg_readers)

harp/schema.py

@@ -11,9 +11,7 @@ def _read_common_registers() -> Registers:
         return parse_yaml_raw_as(Registers, fileIO.read())
 
 
-def read_schema(
-    file: Union[str, PathLike, TextIO], include_common_registers: bool = True
-) -> Model:
+def read_schema(file: Union[str, PathLike, TextIO], include_common_registers: bool = True) -> Model:
     """Read and parse a device schema from the specified file.
 
     Parameters
@@ -40,9 +38,7 @@ def read_schema(
             schema.registers = dict(common.registers, **schema.registers)
             if common.bitMasks:
                 schema.bitMasks = (
-                    common.bitMasks
-                    if schema.bitMasks is None
-                    else dict(common.bitMasks, **schema.bitMasks)
+                    common.bitMasks if schema.bitMasks is None else dict(common.bitMasks, **schema.bitMasks)
                 )
             if common.groupMasks:
                 schema.groupMasks = (

tests/test_io.py

@@ -26,9 +26,7 @@
         expected_error=ValueError,
     ),
     DataFileParam(path="data/write_0.bin", expected_address=0, expected_rows=4),
-    DataFileParam(
-        path="data/write_0.bin", expected_address=0, expected_rows=4, keep_type=True
-    ),
+    DataFileParam(path="data/write_0.bin", expected_address=0, expected_rows=4, keep_type=True),
 ]
 
 
@@ -45,10 +43,7 @@ def test_read(dataFile: DataFileParam):
         )
         assert len(data) == dataFile.expected_rows
         if dataFile.keep_type:
-            assert (
-                MessageType.__name__ in data.columns
-                and data[MessageType.__name__].dtype == "category"
-            )
+            assert MessageType.__name__ in data.columns and data[MessageType.__name__].dtype == "category"
 
         if dataFile.expected_cols:
             for col in dataFile.expected_cols: