Skip to content

Commit 77aaa31

Browse files
committed
Deprecation: Fix ptp --> np.ptp for numpy 2.0.0
1 parent decf198 commit 77aaa31

File tree

4 files changed

+6
-4
lines changed

4 files changed

+6
-4
lines changed

diffsims/generators/diffraction_generator.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -527,7 +527,7 @@ def calculate_ed_data(
527527

528528
# Add z-coordinate
529529
z_range = max(
530-
z_range, coordinates[:, -1].ptp()
530+
z_range, np.ptp(coordinates[:, -1])
531531
) # enforce minimal resolution in reciprocal space
532532
x = [
533533
self.detector[0],

diffsims/simulations/simulation2d.py

+1
Original file line numberDiff line numberDiff line change
@@ -253,6 +253,7 @@ def current_size(self):
253253
return self.rotations.size
254254

255255
def deepcopy(self):
256+
256257
return copy.deepcopy(self)
257258

258259
def _get_transformed_coordinates(

diffsims/tests/utils/test_fourier_transform.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -160,9 +160,9 @@ def test_freq(shape, dX, rX, dY, rY):
160160
if dX[i] is not None:
161161
assert abs(x[i].item(1) - x[i].item(0)) <= dX[i] + 1e-8
162162
if rY[i] is not None:
163-
assert y[i].ptp() >= rY[i] - 1e-8
163+
assert np.ptp(y[i]) >= rY[i] - 1e-8
164164
if rX[i] is not None:
165-
assert x[i].ptp() >= rX[i] - 1e-8
165+
assert np.ptp(x[i]) >= rX[i] - 1e-8
166166
if dY[i] is not None:
167167
assert abs(y[i].item(1) - y[i].item(0)) <= dY[i] + 1e-8
168168

diffsims/utils/discretise_utils.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@
4444
exp,
4545
prod,
4646
)
47+
import numpy as np
4748
from psutil import virtual_memory
4849
import numba
4950

@@ -321,7 +322,7 @@ def rebin(x, loc, r, k, mem):
321322
else:
322323
r = r.copy()
323324
xmin = array([X.item(0) if X.size > 1 else -1e5 for X in x], dtype=x[0].dtype)
324-
nbins = [int(ceil(x[i].ptp() / r[i])) + 1 for i in range(3)]
325+
nbins = [int(ceil(np.ptp(x[i]) / r[i])) + 1 for i in range(3)]
325326
if prod(nbins) * 32 * 10 > mem:
326327
raise MemoryError
327328
Len = zeros(nbins, dtype="i4")

0 commit comments

Comments
 (0)