diff --git a/activitysim/abm/models/util/test/test_vectorize_tour_scheduling.py b/activitysim/abm/models/util/test/test_vectorize_tour_scheduling.py index 3c3cc5c90..8dc7aaa39 100644 --- a/activitysim/abm/models/util/test/test_vectorize_tour_scheduling.py +++ b/activitysim/abm/models/util/test/test_vectorize_tour_scheduling.py @@ -1,10 +1,13 @@ # ActivitySim # See full license in LICENSE.txt. +from __future__ import annotations + import pandas as pd import pandas.testing as pdt from activitysim.abm.models.util.vectorize_tour_scheduling import ( + TourSchedulingSettings, get_previous_tour_by_tourid, vectorize_tour_scheduling, ) @@ -64,7 +67,7 @@ def test_vts(): timetable, tour_segments={"spec": spec}, tour_segment_col=None, - model_settings={}, + model_settings=TourSchedulingSettings(), chunk_size=0, trace_label="test_vts", ) diff --git a/activitysim/abm/models/util/vectorize_tour_scheduling.py b/activitysim/abm/models/util/vectorize_tour_scheduling.py index 88fd2dfd1..b3951dea1 100644 --- a/activitysim/abm/models/util/vectorize_tour_scheduling.py +++ b/activitysim/abm/models/util/vectorize_tour_scheduling.py @@ -981,7 +981,7 @@ def vectorize_tour_scheduling( spec : DataFrame The spec which will be passed to interaction_simulate. (or dict of specs keyed on tour_type if tour_types is not None) - model_settings : LOGSUM_SETTINGS + model_settings : TourSchedulingSettings Returns -------