Skip to content

Commit 61c7a5c

Browse files
committed
Always drop description column
1 parent dd0b540 commit 61c7a5c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

switch_model/wecc/get_inputs/post_process_steps/add_storage.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,13 @@ def fetch_df(tab_name, key, config):
2424
gid = TAB_NAME_GID[tab_name]
2525
url = f"https://docs.google.com/spreadsheet/ccc?key={SHEET_ID}&output=csv&gid={gid}"
2626

27-
df = pd.read_csv(url, index_col=False) \
27+
df: pd.DataFrame = pd.read_csv(url, index_col=False) \
2828
.replace("FALSE", 0) \
2929
.replace("TRUE", 1)
3030

31+
if "description" in df.columns:
32+
df = df.drop("description", axis=1)
33+
3134
if key is not None:
3235
df = filer_by_scenario(df, key, config)
3336
return df

0 commit comments

Comments
 (0)