allow skims in trip mode choice annotate (#857)

ActivitySim · Apr 24, 2024 · ccd0477 · ccd0477
1 parent 67fa08e
commit ccd0477
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/activitysim/abm/models/trip_mode_choice.py b/activitysim/abm/models/trip_mode_choice.py
@@ -371,6 +371,12 @@ def trip_mode_choice(
     state.add_table("trips", trips_df)
 
     if model_settings.annotate_trips:
+        # need to update locals_dict to access skims that are the same .shape as trips table
+        locals_dict = {}
+        locals_dict.update(constants)
+        simulate.set_skim_wrapper_targets(trips_merged, skims)
+        locals_dict.update(skims)
+        locals_dict["timeframe"] = "trip"
         annotate.annotate_trips(state, model_settings, trace_label, locals_dict)
 
     if state.settings.trace_hh_id: