Typing fixes

Author: hexane360

atomlib/vec.py

@@ -8,7 +8,7 @@
 from numpy.typing import ArrayLike, NDArray
 
 # re-export to_vec3
-from .types import to_vec3, ScalarT
+from .types import to_vec3, ScalarT, NumT
 
 
 def dot(v1: ArrayLike, v2: ArrayLike, axis: int = -1, keepdims: bool = True) -> NDArray[numpy.floating]:
@@ -93,7 +93,7 @@ def polygon_solid_angle(poly: ArrayLike, pts: t.Optional[ArrayLike] = None,
     # normalize polygon points to unit sphere
     numpy.divide(poly, numpy.linalg.norm(poly, axis=-1, keepdims=True), out=poly)
 
-    def _dot(v1: NDArray[numpy.float64], v2: NDArray[numpy.float64]) -> NDArray[numpy.float64]:
+    def _dot(v1: NDArray[NumT], v2: NDArray[NumT]) -> NDArray[NumT]:
         return numpy.add.reduce(v1 * v2, axis=-1)
 
     # next and previous points in polygon

pyproject.toml

@@ -30,8 +30,8 @@ classifiers = [
 requires-python = ">=3.9"
 dependencies = [
     "click~=8.1",  # for cli
-    "numpy>=1.22,<2.3.0",  # tested on 2.0.0
-    "scipy~=1.8",
+    "numpy>=1.22,<2.6.0",  # tested on 2.3.1
+    "scipy~=1.8",          # tested on 1.16.0
     "polars~=1.31.0",
     "matplotlib~=3.5",
     "requests~=2.28",