Assign a type to get_guide_catalog cats to plot

sot · Feb 21, 2019 · a5cc680 · a5cc680
1 parent edee25b
commit a5cc680
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):