Merge pull request #819 from wsp-sag/time-period-cat-legacy-mode

Time period to categorical in legacy mode

activitysim/core/skim_dictionary.py

@@ -1,6 +1,7 @@
 # ActivitySim
 # See full license in LICENSE.txt.
 from __future__ import annotations
+from collections import OrderedDict
 
 import logging
 from builtins import object, range
@@ -9,6 +10,7 @@
 import pandas as pd
 
 from activitysim.core import workflow
+from activitysim.core.exceptions import StateAccessError
 
 logger = logging.getLogger(__name__)
 
@@ -165,6 +167,20 @@ def __init__(self, state, skim_tag, skim_info, skim_data):
         self.skim_info = skim_info
         self.usage = set()  # track keys of skims looked up
 
+        try:
+            self.time_label_dtype = pd.api.types.CategoricalDtype(
+                list(
+                    OrderedDict.fromkeys(
+                        state.network_settings.skim_time_periods.labels
+                    )
+                ),
+                ordered=True,
+            )
+        except StateAccessError:
+            logger.info(
+                f"Cannot access state.network_settings.skim_time_periods.labels. SkimDict.time_label_dtype is not set"
+            )
+
         self.offset_mapper = self._offset_mapper(
             state
         )  # (in function so subclass can override)