TYP: Fix new static typing failures

has2k1 · Dec 10, 2024 · 67c7651 · 67c7651
1 parent f574e1d
commit 67c7651
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 6 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.7.0
+    rev: v0.8.2
     hooks:
       - id: ruff
         args: [--select=I001, --fix]

diff --git a/doc/changelog.rst b/doc/changelog.rst
@@ -1,6 +1,11 @@
 Changelog
 =========
 
+v0.13.1
+-------
+*not-yet-released*
+
+
 v0.13.0
 -------
 *2024-10-24*

diff --git a/mizani/bounds.py b/mizani/bounds.py
@@ -365,7 +365,7 @@ def censor(
     if bool_idx.any():
         if res.dtype == int:
             res = res.astype(float)
-        res[bool_idx] = null
+        res[bool_idx] = null  # pyright: ignore[reportCallIssue,reportArgumentType]
     return res
 
 

diff --git a/mizani/transforms.py b/mizani/transforms.py
@@ -648,7 +648,7 @@ def transform(self, x: FloatArrayLike) -> NDArrayFloat:
         p, offset = self.p, self.offset
 
         if np.abs(self.p) < 1e-7:
-            return np.sign(x) * np.log(np.abs(x) + offset)
+            return np.sign(x) * np.log(np.abs(x) + offset)  # pyright: ignore[reportReturnType]
         else:
             return np.sign(x) * ((np.abs(x) + offset) ** p - 1) / p
 
@@ -916,7 +916,7 @@ class symlog_trans(trans):
     breaks_func: BreaksFunction = breaks_symlog()
 
     def transform(self, x: FloatArrayLike) -> NDArrayFloat:
-        return np.sign(x) * np.log1p(np.abs(x))
+        return np.sign(x) * np.log1p(np.abs(x))  # pyright: ignore[reportReturnType]
 
     def inverse(self, x: FloatArrayLike) -> NDArrayFloat:
         return np.sign(x) * (np.exp(np.abs(x)) - 1)  # type: ignore

diff --git a/mizani/typing.py b/mizani/typing.py
@@ -28,7 +28,7 @@
 
     # Arrays (strictly numpy)
     NDArrayAny: TypeAlias = NDArray[Any]
-    NDArrayFloat: TypeAlias = NDArray[np.float64]
+    NDArrayFloat: TypeAlias = NDArray[np.floating]
     NDArrayDatetime: TypeAlias = NDArray[np.datetime64]
 
     # Panda Series

diff --git a/pyproject.toml b/pyproject.toml
@@ -65,7 +65,7 @@ dev = [
 ]
 
 typing = [
-    "pyright==1.1.386",
+    "pyright==1.1.390",
     "pandas-stubs",
 ]