Skip to content

Commit

Permalink
Merge pull request #585 from braingram/poly1d
Browse files Browse the repository at this point in the history
Numpy 2.0 fixes:poly1d and NaN
  • Loading branch information
BradleySappington authored Oct 13, 2023
2 parents eee741a + b2f1c3f commit f373e0d
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
8 changes: 4 additions & 4 deletions poppy/accel_math.py
Original file line number Diff line number Diff line change
Expand Up @@ -387,7 +387,7 @@ def benchmark_fft(npix=2048, iterations=20, double_precision=True):
time_fftw = timer.timeit(number=iterations) / iterations
print(" {:.3f} s".format(time_fftw))
else:
time_fftw = np.NaN
time_fftw = np.nan

if poppy.accel_math._NUMEXPR_AVAILABLE:
print("Timing performance with Numexpr + FFTW:")
Expand All @@ -396,7 +396,7 @@ def benchmark_fft(npix=2048, iterations=20, double_precision=True):
time_numexpr = timer.timeit(number=iterations) / iterations
print(" {:.3f} s".format(time_numexpr))
else:
time_numexpr = np.NaN
time_numexpr = np.nan

if poppy.accel_math._MKLFFT_AVAILABLE:
print("Timing performance with Numexpr + MKL:")
Expand All @@ -407,7 +407,7 @@ def benchmark_fft(npix=2048, iterations=20, double_precision=True):
time_mkl = timer.timeit(number=iterations) / iterations
print(" {:.3f} s".format(time_mkl))
else:
time_mkl = np.NaN
time_mkl = np.nan

if poppy.accel_math._OPENCL_AVAILABLE:
print("Timing performance with OpenCL:")
Expand All @@ -416,7 +416,7 @@ def benchmark_fft(npix=2048, iterations=20, double_precision=True):
time_opencl = timer.timeit(number=iterations) / iterations
print(" {:.3f} s".format(time_opencl))
else:
time_opencl = np.NaN
time_opencl = np.nan


poppy.conf.use_mkl, poppy.conf.use_fftw, poppy.conf.use_numexpr, poppy.conf.use_cupy,\
Expand Down
2 changes: 1 addition & 1 deletion poppy/optics.py
Original file line number Diff line number Diff line change
Expand Up @@ -566,7 +566,7 @@ def get_transmission(self, wave):
else:
raise NotImplemented("No defined NIRCam wedge BLC mask for that wavelength? ")

sigmas = scipy.poly1d(polyfitcoeffs)(scalefact)
sigmas = numpy.poly1d(polyfitcoeffs)(scalefact)

sigmar = sigmas * np.abs(y)
sigmar.clip(np.finfo(sigmar.dtype).tiny, out=sigmar) # avoid divide by zero -> NaNs
Expand Down

0 comments on commit f373e0d

Please sign in to comment.