Skip to content

Commit

Permalink
Adapted wedge test in mu_sigma estimate
Browse files Browse the repository at this point in the history
  • Loading branch information
VChristiaens committed Nov 25, 2024
1 parent 6158995 commit 1396286
Showing 1 changed file with 8 additions and 7 deletions.
15 changes: 8 additions & 7 deletions vip_hci/fm/negfc_fmerit.py
Original file line number Diff line number Diff line change
Expand Up @@ -924,13 +924,14 @@ def _estimate_mu_sigma_3d(cube, angs, ncomp, annulus_width, aperture_radius,
theta_ini = (theta_guess + delta_theta) % 360
theta_fin = theta_ini + (360 - 2 * delta_theta)
wedge = (theta_ini, theta_fin)
if len(wedge) == 2:
if wedge[0] > wedge[1]:
msg = "2nd value of wedge smaller than first one => +360"
print(msg)
wedge = (wedge[0], wedge[1] + 360)
else:
raise TypeError("Wedge should have exactly 2 values")
if wedge is not None:
if len(wedge) == 2:
if wedge[0] > wedge[1]:
msg = "2nd value of wedge smaller than first one => +360"
print(msg)
wedge = (wedge[0], wedge[1] + 360)
else:
raise TypeError("Wedge should have exactly 2 values")

# annulus to estimate mu & sigma should encompass the companion location
indices = get_annular_wedge(pca_res, inner_radius=radius_int,
Expand Down

0 comments on commit 1396286

Please sign in to comment.