From f92338a5b0b0aa7e40bbeae670f50db267101dc8 Mon Sep 17 00:00:00 2001 From: Ben RdO Date: Wed, 16 Sep 2020 19:34:09 -0700 Subject: [PATCH] fix: consider full list of resources --- postreise/plot/analyze_pg.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/postreise/plot/analyze_pg.py b/postreise/plot/analyze_pg.py index 908c6c7c..44c6644a 100644 --- 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]