Skip to content

Commit baf44ca

Browse files
committed
replace additional np.ndarray occurences in type hints
1 parent 678a4b5 commit baf44ca

2 files changed

Lines changed: 11988 additions & 5 deletions

File tree

yt/testing.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def assert_rel_equal(a1, a2, decimals, err_msg="", verbose=True):
9292

9393
# tested: volume integral is 1.
9494
def cubicspline_python(
95-
x: float | np.ndarray,
95+
x: float | NDArray[np.floating],
9696
) -> np.ndarray:
9797
"""
9898
cubic spline SPH kernel function for testing against more
@@ -157,11 +157,11 @@ def integrate_kernel(
157157

158158

159159
def distancematrix(
160-
pos3_i0: np.ndarray,
161-
pos3_i1: np.ndarray,
160+
pos3_i0: NDArray[np.floating],
161+
pos3_i1: NDArray[np.floating],
162162
periodic: tuple[bool, bool, bool] = (True,) * 3,
163-
periods: np.ndarray = _zeroperiods,
164-
) -> np.ndarray:
163+
periods: NDArray[np.floating] = _zeroperiods,
164+
) -> NDArray[np.floating]:
165165
"""
166166
Calculates the distances between two arrays of points.
167167

0 commit comments

Comments
 (0)