Merge pull request #41 from Stoops-ML/helper_types

FIX: output types of helpers.py: deg_angular_difference(), rad_angula…
Stoops-ML · Oct 8, 2024 · 7d3d440 · 7d3d440
2 parents 8c46a3d + 46fb31b
commit 7d3d440
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/transforms84/helpers.pyi b/src/transforms84/helpers.pyi
@@ -17,7 +17,10 @@ def deg_angular_difference(
         ],
     ],
     smallest_angle: bool,
-) -> float: ...
+) -> Union[
+    float,
+    npt.NDArray[Union[np.float32, np.float64, np.int8, np.int16, np.int32, np.int64]],
+]: ...
 def rad_angular_difference(
     angle1: Union[
         float,
@@ -32,7 +35,10 @@ def rad_angular_difference(
         ],
     ],
     smallest_angle: bool,
-) -> float: ...
+) -> Union[
+    float,
+    npt.NDArray[Union[np.float32, np.float64, np.int8, np.int16, np.int32, np.int64]],
+]: ...
 def RRM2DDM(
     rrm_position: npt.NDArray[
         Union[np.float32, np.float64, np.int8, np.int16, np.int32, np.int64]
@@ -63,4 +69,7 @@ def wrap(
         int,
         npt.NDArray[Union[np.float32, np.float64]],
     ],
-): ...
+) -> Union[
+    float,
+    npt.NDArray[Union[np.float32, np.float64, np.int8, np.int16, np.int32, np.int64]],
+]: ...