fix tour schedule test

ActivitySim · Nov 10, 2023 · b184ae0 · b184ae0
1 parent 4d05c28
commit b184ae0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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
     -------