Update nb_funcs.py

JarronL · Apr 1, 2020 · 34efc79 · 34efc79
1 parent c1910d1
commit 34efc79
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/pynrc/nb_funcs.py b/pynrc/nb_funcs.py
@@ -869,6 +869,11 @@ def plot_planet_patches(ax, obs, age=10, entropy=13, mass_list=[10,5,2,1], av_va
 
     import matplotlib.patches as mpatches
 
+    # Don't plot anything if 
+    if mass_list is None:
+        _log.info("mass_list=None; Not plotting planet patch locations.")
+        return
+
     xlim = ax.get_xlim()
 
 
@@ -1177,8 +1182,6 @@ def do_plot_contrasts2(key1, key2, curves_all, nsig, obs_dict, wfe_list, age, sa
     ax, ax2, ax3 = plot_contrasts(curves, nsig, wfe_list, obs=obs, sat_rad=sat_rad,
                                   ax=ax, colors=c1, xr=xr, yr=yr, return_axes=True)
     axes1_all = [ax, ax2, ax3]
-    # Planet mass locations
-    plot_planet_patches(ax, obs, age=age, update_title=True, av_vals=av_vals, linder=linder_models, **kwargs)
 
     if key2 is not None:
         k = key2
@@ -1188,6 +1191,9 @@ def do_plot_contrasts2(key1, key2, curves_all, nsig, obs_dict, wfe_list, age, sa
         plot_contrasts(curves, nsig, wfe_list, obs=obs, sat_rad=sat_rad, 
                        ax=ax, xr=xr, yr=yr, colors=c2)
 
+    # Planet mass locations
+    plot_planet_patches(ax, obs, age=age, update_title=True, av_vals=av_vals, linder=linder_models, **kwargs)
+
 
 
     # Right plot (Converted to MJup/MEarth)