Small updates to solve failing checks in previous commit

rpoleski · Nov 16, 2023 · 6ac184c · 6ac184c
1 parent 9704c80
commit 6ac184c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/example_16/ulens_model_fit.py b/examples/example_16/ulens_model_fit.py
@@ -3138,7 +3138,7 @@ def _second_Y_axis_get_fluxes(self, ylim):
         A_min = (flux_min - blend_flux) / total_source_flux
         A_max = (flux_max - blend_flux) / total_source_flux
 
-        return (A_min, A_max, (total_source_flux, blend_flux))
+        return (A_min, A_max, [total_source_flux, blend_flux])
 
     def _second_Y_axis_optimal(self, ax2, A_min, A_max):
         """
@@ -3169,7 +3169,7 @@ def _second_Y_axis_warnings(self, flux, labels, A_values, A_min, A_max):
         Issue warnings for negative flux or bad range of magnificaitons
         """
         if np.any(flux < 0.):
-            mask = flux > 0.
+            mask = (flux > 0.)
             flux = flux[mask]
             labels = [l for (l, m) in zip(labels, mask) if m]
             msg = ("\n\n{:} label/s on the second Y scale will not be shown "