try to fix mypy issue

resgroup · Jan 5, 2025 · cf7e49d · cf7e49d
1 parent 36d4611
commit cf7e49d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -79,8 +79,6 @@ show-fixes = true
 [tool.ruff.lint]
 select = ["ALL"] # https://beta.ruff.rs/docs/rules/
 ignore = [
-    "ANN101", # `self` doesn't need annotations
-    "ANN102", # `cls` doesn't need annotations
     "ANN204", # `__init__` doesn't need annotations
     "S301", # `pickle` and modules that wrap it can be unsafe when used to deserialize untrusted data, possible security issue
     "D", # docstring checks

diff --git a/wind_up/scada_funcs.py b/wind_up/scada_funcs.py
@@ -3,7 +3,7 @@
 import datetime as dt
 import logging
 import warnings
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 import numpy as np
 import pandas as pd
@@ -249,7 +249,7 @@ def filter_rpm_or_pt_curve(
     x_col: str,
     y_col: str,
     y_centile: float,
-    x_bin_edges: list[float],
+    x_bin_edges: list[np.floating[Any]],
     reject_high: bool,
     y_margin: float,
     filter_last_bin: bool = True,