Improve type hints.

precice · Nov 27, 2024 · 4ded055 · 4ded055
1 parent d27d64f
commit 4ded055
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/oscillator-overlap/solver-python/mypy.ini b/oscillator-overlap/solver-python/mypy.ini
@@ -0,0 +1,7 @@
+[mypy]
+
+[mypy-scipy.*]
+ignore_missing_imports = True
+
+[mypy-precice.*]
+ignore_missing_imports = True
diff --git a/oscillator-overlap/solver-python/problemDefinition.py b/oscillator-overlap/solver-python/problemDefinition.py
@@ -1,5 +1,6 @@
 import numpy as np
 from numpy.linalg import eig
+from numpy.typing import NDArray
 from typing import Callable
 
 
@@ -23,8 +24,8 @@ class Mass:
     m: float
     u0: float
     v0: float
-    u_analytical: Callable[[float | np.ndarray], float | np.ndarray]
-    v_analytical: Callable[[float | np.ndarray], float | np.ndarray]
+    u_analytical: Callable[[float | NDArray], float | NDArray]
+    v_analytical: Callable[[float | NDArray], float | NDArray]
 
 
 class MassLeft(Mass):

diff --git a/oscillator/solver-python/problemDefinition.py b/oscillator/solver-python/problemDefinition.py
@@ -1,5 +1,6 @@
 import numpy as np
 from numpy.linalg import eig
+from numpy.typing import NDArray
 from typing import Callable
 
 
@@ -23,8 +24,8 @@ class Mass:
     m: float
     u0: float
     v0: float
-    u_analytical: Callable[[float | np.ndarray], float | np.ndarray]
-    v_analytical: Callable[[float | np.ndarray], float | np.ndarray]
+    u_analytical: Callable[[float | NDArray], float | NDArray]
+    v_analytical: Callable[[float | NDArray], float | NDArray]
 
 
 class MassLeft(Mass):