File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1313import pytest
1414
1515from activitysim .core import random , tracing , workflow
16- from activitysim .core .exceptions import CheckpointNameNotFoundError , TableSlicingError
16+ from activitysim .core .exceptions import (
17+ CheckpointNameNotFoundError ,
18+ DuplicateWorkflowNameError ,
19+ )
1720
1821# set the max households for all tests (this is to limit memory use on travis)
1922HOUSEHOLDS_SAMPLE_SIZE = 50
@@ -236,7 +239,7 @@ def test_mini_pipeline_run2():
236239 regress_mini_auto (state )
237240
238241 # try to run a model already in pipeline
239- with pytest .raises (TableSlicingError ) as excinfo :
242+ with pytest .raises (DuplicateWorkflowNameError ) as excinfo :
240243 state .run .by_name ("auto_ownership_simulate" )
241244 assert "run model 'auto_ownership_simulate' more than once" in str (excinfo .value )
242245
Original file line number Diff line number Diff line change 77from datetime import timedelta
88
99from activitysim .core import tracing
10- from activitysim .core .exceptions import DuplicateWorkflowNameError , TableSlicingError
10+ from activitysim .core .exceptions import DuplicateWorkflowNameError
1111from activitysim .core .workflow .accessor import FromState , StateAccessor
1212from activitysim .core .workflow .checkpoint import (
1313 CHECKPOINT_NAME ,
@@ -265,7 +265,7 @@ def _pre_run_step(self, model_name: str) -> bool | None:
265265 if model_name in checkpointed_models :
266266 if self ._obj .settings .duplicate_step_execution == "error" :
267267 checkpointed_model_bullets = "\n - " .join (checkpointed_models )
268- raise TableSlicingError (
268+ raise DuplicateWorkflowNameError (
269269 f"Checkpointed Models:\n - { checkpointed_model_bullets } \n "
270270 f"Cannot run model '{ model_name } ' more than once"
271271 )
You can’t perform that action at this time.
0 commit comments