From c9b7f188136bfa2510ef2c33131e7fd9eb81debd Mon Sep 17 00:00:00 2001 From: LIU ZHANWEI <50036800+Zhanwei-Liu@users.noreply.github.com> Date: Fri, 4 Aug 2023 19:48:37 +0800 Subject: [PATCH] Update load_data.py fix a bug to ignore static sheet when not including hydro --- prepshot/load_data.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/prepshot/load_data.py b/prepshot/load_data.py index da4b0d1..45e20d9 100644 --- a/prepshot/load_data.py +++ b/prepshot/load_data.py @@ -87,7 +87,8 @@ def get_attr(para): None """ para["year"] = sorted(list(para["discount_factor"].keys())) - para["stcd"] = list({i[1] for i in para["static"].keys()}) + if "static" in para.keys(): + para["stcd"] = list({i[1] for i in para["static"].keys()}) para["hour"] = sorted({i[3] for i in para["demand"].keys() if isinstance(i[3], int)}) para["month"] = sorted({i[2] for i in para["demand"].keys() if isinstance(i[2], int)}) para["zone"] = list({i[0] for i in para["demand"].keys()})