test: fix numpy array return type

use numpy.typing.NDArray to type array and dtype correctly
readyplayerme · Nov 8, 2023 · b4e819f · b4e819f
1 parent e6db576
commit b4e819f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -45,6 +45,7 @@ repos:
     rev: 'v1.5.1'
     hooks:
       - id: mypy
+        additional_dependencies: [numpy]
 
   - repo: https://github.com/espressif/conventional-precommit-linter
     rev: v1.2.1

diff --git a/src/readyplayerme/meshops/mesh.py b/src/readyplayerme/meshops/mesh.py
@@ -4,6 +4,7 @@
 from typing import TypeAlias
 
 import numpy as np
+import numpy.typing as npt
 import trimesh
 
 # Abstraction for the mesh type.
@@ -37,7 +38,7 @@ def read_glb(filename: str | Path) -> Mesh:
     return trimesh.load(filename, process=False, force="mesh")
 
 
-def get_border_vertices(mesh: Mesh) -> np.array:
+def get_border_vertices(mesh: Mesh) -> npt.NDArray[np.int64]:
     """Return the indices of the vertices on the borders of a mesh.
 
     :param mesh: The mesh to get the border vertices from.