Fix active skim columns in travel time reporter

SANDAG · Oct 29, 2024 · bfd0ca2 · bfd0ca2
1 parent b2b3eda
commit bfd0ca2
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/src/main/python/TravelTimeReporter.py b/src/main/python/TravelTimeReporter.py
@@ -107,14 +107,22 @@ def read_active_skims(self):
         Reads active skims into memory as data frames
         """
         for skim_name in self.settings["active_skim_files"]:
-            self.skims[skim_name] = pd.read_csv(
-                        os.path.join(
-                        self.model_run,
-                        "output",
-                        "skims",
-                        self.settings["active_skim_files"][skim_name]
+            active_skims = pd.read_csv(
+                os.path.join(
+                    self.model_run,
+                    "output",
+                    "skims",
+                    self.settings["active_skim_files"][skim_name]
                 )
-            ).set_index(
+            )
+            if not "i" in active_skims.columns:
+                active_skims = active_skims.rename(
+                    columns = {
+                        "OMAZ": "i",
+                        "DMAZ": "j",
+                    }
+                )
+            self.skims[skim_name] = active_skims.set_index(
                 ["i", "j"]
             )