bug fix with pm.math -> tensor.math (argmin)

hposborn · Jul 8, 2024 · 9185650 · 9185650
1 parent c11b1ce
commit 9185650
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/MonoTools/fit.py b/MonoTools/fit.py
@@ -2153,8 +2153,8 @@ def create_orbit(pl, Rs, rho_S, pers, t0s, bs, n_marg=1, eccs=None, omegas=None)
                 elif self.interpolate_v_prior and pl in self.trios+self.duos+self.monos:
                     #  We only need to create one orbit if we're not marginalising over N periods
                     #      (i.e. when we only have the lightcurve and we're interpolating a velocity prior)
-                    orbits[pl] = create_orbit(pl, Rs, rho_S, pers[pl][pm.math.argmin(min_eccs[pl])], t0s[pl], bs[pl], n_marg=1,
-                                              omegas=omegas[pl][pm.math.argmin(min_eccs[pl])], eccs=pm.math.min(min_eccs[pl]))
+                    orbits[pl] = create_orbit(pl, Rs, rho_S, pers[pl][tensor.math.argmin(min_eccs[pl])], t0s[pl], bs[pl], n_marg=1,
+                                              omegas=omegas[pl][tensor.math.argmin(min_eccs[pl])], eccs=pm.math.min(min_eccs[pl]))
                     light_curves[pl] = gen_lc(orbits[pl], rpls[pl]/109.2, 1, mask=None,
                                               prefix=pl+'_', make_deterministic=True)
                 else: