chore: drop id col and reset when needed

Breakthrough-Energy · Jan 6, 2021 · 6149e15 · 6149e15
1 parent 79383c9
commit 6149e15
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/powersimdata/data_access/csv_store.py b/powersimdata/data_access/csv_store.py
@@ -43,7 +43,7 @@ def _parse_csv(self, file_object):
         :return: (*pandas.DataFrame*) -- the specified file as a data frame.
         """
         table = pd.read_csv(file_object)
-        table.set_index("id", inplace=True, drop=False)
+        table.set_index("id", inplace=True)
         table.fillna("", inplace=True)
         return table.astype(str)
 

diff --git a/powersimdata/data_access/scenario_list.py b/powersimdata/data_access/scenario_list.py
@@ -141,7 +141,11 @@ def err_message(text):
             print(f"Duplicate ids: {dupes}")
             print("Use id to access scenario")
         else:
-            return scenario.to_dict("records", into=OrderedDict)[0]
+            return (
+                scenario.reset_index()
+                .astype({"id": "str"})
+                .to_dict("records", into=OrderedDict)[0]
+            )
 
     def add_entry(self, scenario_info):
         """Adds scenario to the scenario list file on server.