Added TableIndexError

Author: JoeJimFlood

activitysim/core/exceptions.py

@@ -77,8 +77,14 @@ class TableSlicingError(RuntimeError):
 class InputPopulationError(RuntimeError):
     """An issue with the input population was found."""
 
+
 class SubprocessError(RuntimeError):
     """An error occurred in a subprocess."""
 
+
 class SegmentedSpecificationError(RuntimeError):
-    """An error was caused by creating an invalid spec table for a segmented model component."""
+    """An error was caused by creating an invalid spec table for a segmented model component."""
+
+
+class TableIndexError(RuntimeError):
+    """An error related to the index of a table in the pipeline."""

activitysim/core/logit.py

@@ -11,7 +11,7 @@
 from activitysim.core import tracing, workflow
 from activitysim.core.choosing import choice_maker
 from activitysim.core.configuration.logit import LogitNestSpec
-from activitysim.core.exceptions import InvalidTravelError, ModelConfigurationError
+from activitysim.core.exceptions import InvalidTravelError, ModelConfigurationError, TableIndexError
 
 logger = logging.getLogger(__name__)
 
@@ -361,11 +361,11 @@ def interaction_dataset(
 
     """
     if not choosers.index.is_unique:
-        raise RuntimeError(
+        raise TableIndexError(
             "ERROR: choosers index is not unique, " "sample will not work correctly"
         )
     if not alternatives.index.is_unique:
-        raise RuntimeError(
+        raise TableIndexError(
             "ERROR: alternatives index is not unique, " "sample will not work correctly"
         )
 

activitysim/core/random.py

@@ -10,7 +10,7 @@
 import pandas as pd
 
 from activitysim.core.util import reindex
-from activitysim.core.exceptions import DuplicateLoadableObjectError
+from activitysim.core.exceptions import DuplicateLoadableObjectError, TableIndexError
 
 from .tracing import print_elapsed_time
 
@@ -396,7 +396,7 @@ def get_channel_for_df(self, df):
 
         channel_name = self.index_to_channel.get(df.index.name, None)
         if channel_name is None:
-            raise RuntimeError("No channel with index name '%s'" % df.index.name)
+            raise TableIndexError("No channel with index name '%s'" % df.index.name)
         return self.channels[channel_name]
 
     # step handling

activitysim/core/skim_dictionary.py

@@ -10,7 +10,7 @@
 import pandas as pd
 
 from activitysim.core import workflow
-from activitysim.core.exceptions import StateAccessError
+from activitysim.core.exceptions import StateAccessError, TableIndexError
 
 logger = logging.getLogger(__name__)
 
@@ -907,7 +907,7 @@ def get(self, row_ids, col_ids):
             not_in_skim = not_in_skim.values
             logger.warning(f"row_ids: {row_ids[not_in_skim]}")
             logger.warning(f"col_ids: {col_ids[not_in_skim]}")
-            raise RuntimeError(
+            raise TableIndexError(
                 f"DataFrameMatrix: {not_in_skim.sum()} row_ids of {len(row_ids)} not in skim."
             )