Linting

Author: bruno-f-cruz

harp/io.py

@@ -2,9 +2,10 @@
 from enum import IntEnum
 from os import PathLike
 from typing import Any, BinaryIO, Optional, Union
-from pandas._typing import Axes
+
 import numpy as np
 import pandas as pd
+from pandas._typing import Axes
 
 REFERENCE_EPOCH = datetime(1904, 1, 1)
 """The reference epoch for UTC harp time."""

harp/model.py

@@ -6,16 +6,10 @@
 
 from enum import Enum
 from typing import Dict, List, Optional, Union
-from typing_extensions import Annotated
 
-from pydantic import (
-    BaseModel,
-    BeforeValidator,
-    ConfigDict,
-    Field,
-    RootModel,
-    field_serializer,
-)
+from pydantic import (BaseModel, BeforeValidator, ConfigDict, Field, RootModel,
+                      field_serializer)
+from typing_extensions import Annotated
 
 
 class PayloadType(str, Enum):

harp/reader.py

@@ -1,20 +1,23 @@
 import os
-import requests
-from deprecated import deprecated
+from collections import UserDict
+from dataclasses import dataclass
+from datetime import datetime
+from functools import partial
 from math import log2
 from os import PathLike
 from pathlib import Path
-from datetime import datetime
-from functools import partial
-from dataclasses import dataclass
+from typing import (Any, BinaryIO, Callable, Iterable, Mapping, Optional,
+                    Protocol, Union)
+
+import requests
+from deprecated import deprecated
 from numpy import dtype
 from pandas import DataFrame, Series
-from typing import Any, BinaryIO, Callable, Iterable, Mapping, Optional, Protocol, Union
-from collections import UserDict
 from pandas._typing import Axes
+
 from harp import __version__
-from harp.model import BitMask, GroupMask, Model, PayloadMember, Register
 from harp.io import MessageType, read
+from harp.model import BitMask, GroupMask, Model, PayloadMember, Register
 from harp.schema import read_schema
 
 
@@ -31,8 +34,7 @@ def __call__(
         file: Optional[Union[str, bytes, PathLike[Any], BinaryIO]] = None,
         epoch: Optional[datetime] = None,
         keep_type: bool = False,
-    ) -> DataFrame:
-        ...
+    ) -> DataFrame: ...
 
 
 class RegisterReader:
@@ -310,9 +312,12 @@ def from_dataset(
                 base_path=path,
                 include_common_registers=include_common_registers,
                 epoch=epoch,
-                keep_type=keep_type)
+                keep_type=keep_type,
+            )
         else:
-            raise ValueError("The dataset must be a directory containing a device.yml file.")
+            raise ValueError(
+                "The dataset must be a directory containing a device.yml file."
+            )
 
 
 def _compose_parser(
@@ -462,7 +467,10 @@ def parser(df: DataFrame):
     reader = partial(reader, columns=[name])
     return RegisterReader(register, reader)
 
-@deprecated("This function is deprecated. Use DeviceReader.from_file, DeviceReader.from_url, DeviceReader.from_str, and DeviceReader.from_model instead.")
+
+@deprecated(
+    "This function is deprecated. Use DeviceReader.from_file, DeviceReader.from_url, DeviceReader.from_str, and DeviceReader.from_model instead."
+)
 def create_reader(
     device: Union[str, PathLike, Model],
     include_common_registers: bool = True,

harp/schema.py

@@ -1,8 +1,10 @@
+from importlib import resources
 from os import PathLike
 from typing import TextIO, Union
-from harp.model import Model, Registers
+
 from pydantic_yaml import parse_yaml_raw_as
-from importlib import resources
+
+from harp.model import Model, Registers
 
 
 def _read_common_registers() -> Registers: