Merge pull request #286 from sot/plot_guide_cats

Assign a type to get_guide_catalog cats to plot
sot · Feb 21, 2019 · bc00272 · bc00272
2 parents 4e8c208 + a5cc680
commit bc00272
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/proseco/core.py b/proseco/core.py
@@ -678,9 +678,9 @@ def get_catalog_for_plot(self):
         of changing all the `type` column values to self.catalog_type (e.g. 'ACQ').
 
         """
-        catalog = self.as_array()
         if self.catalog_type:
-            catalog['type'] = self.catalog_type
+            self['type'] = self.catalog_type
+        catalog = self.as_array()
         return catalog
 
     def plot(self, ax=None, **kwargs):