Merge pull request #290 from sot/set_pickle_stars

Set stars for plot() if needed
sot · Aug 10, 2019 · 9d8064b · 9d8064b
2 parents ad6f861 + b235058
commit 9d8064b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/proseco/core.py b/proseco/core.py
@@ -688,6 +688,8 @@ def plot(self, ax=None, **kwargs):
         :param kwargs: other keyword args for plot_stars
         """
         kwargs.setdefault('catalog', self.get_catalog_for_plot())
+        if self.stars is None:
+            self.set_stars()
         kwargs.setdefault('stars', self.stars)
         kwargs.setdefault('bad_stars', self.bad_stars_mask)
         return super().plot(ax, **kwargs)