fix: consider full list of resources

Breakthrough-Energy · Sep 17, 2020 · f92338a · f92338a
1 parent 775a92e
commit f92338a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/postreise/plot/analyze_pg.py b/postreise/plot/analyze_pg.py
@@ -5,7 +5,6 @@
 import matplotlib.dates as mdates
 from pandas.plotting import scatter_matrix
 
-from postreise.analyze.helpers import get_resources_in_grid
 from powersimdata.network.usa_tamu.constants.plants import type2color, type2label
 
 
@@ -378,7 +377,7 @@ def _get_chart(self, zone):
                     )
 
             t2l = type2label.copy()
-            for t in get_resources_in_grid(self.grid):
+            for t in type2label.keys():
                 if t not in pg_groups.columns:
                     del t2l[t]
 
@@ -505,7 +504,7 @@ def _get_stacked(self, zone):
                 ax = fig.gca()
 
             t2l = type2label.copy()
-            for t in get_resources_in_grid(self.grid):
+            for t in type2label.keys():
                 if t not in pg_stack.columns:
                     del t2l[t]