Merge pull request spacetelescope#569 from mfeldt/patch-1

Update zernike.py
mperrin · May 31, 2023 · 132ea76 · 132ea76
2 parents f31dbba + eedfde0
commit 132ea76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/poppy/zernike.py b/poppy/zernike.py
@@ -765,7 +765,7 @@ def arbitrary_basis(aperture, nterms=15, rho=None, theta=None, outside=np.nan):
         "only square aperture arrays are supported"
 
     # any pixels with zero or NaN in the aperture are outside the area
-    apmask = (xp.isfinite(aperture) & (aperture > 0))
+    apmask = (xp.isfinite(xp.asarray(aperture)) & (xp.asarray(aperture) > 0))
     apmask_float = xp.asarray(apmask, float)
     A = apmask.sum()