diff --git a/activitysim/abm/models/location_choice.py b/activitysim/abm/models/location_choice.py index 937721c05..a7e4f0342 100644 --- a/activitysim/abm/models/location_choice.py +++ b/activitysim/abm/models/location_choice.py @@ -17,6 +17,7 @@ ) from activitysim.core.interaction_sample import interaction_sample from activitysim.core.interaction_sample_simulate import interaction_sample_simulate +from activitysim.core.util import reindex # import multiprocessing @@ -157,6 +158,8 @@ def _location_sample( "orig_col_name": skims.orig_key, # added for sharrow flows "dest_col_name": skims.dest_key, # added for sharrow flows "timeframe": "timeless", + "reindex": reindex, + "land_use": state.get_dataframe("land_use"), } locals_d.update(model_settings.CONSTANTS or {}) @@ -620,6 +623,8 @@ def run_location_simulate( "orig_col_name": skims.orig_key, # added for sharrow flows "dest_col_name": skims.dest_key, # added for sharrow flows "timeframe": "timeless", + "reindex": reindex, + "land_use": state.get_dataframe("land_use"), } locals_d.update(model_settings.CONSTANTS or {}) diff --git a/activitysim/abm/models/trip_destination.py b/activitysim/abm/models/trip_destination.py index 9b7983c87..6dea50980 100644 --- a/activitysim/abm/models/trip_destination.py +++ b/activitysim/abm/models/trip_destination.py @@ -193,6 +193,7 @@ def _destination_sample( "size_terms": size_term_matrix, "size_terms_array": size_term_matrix.df.to_numpy(), "timeframe": "trip", + "land_use": state.get_dataframe("land_use"), } ) locals_dict.update(skims) @@ -925,6 +926,7 @@ def trip_destination_simulate( "size_terms": size_term_matrix, "size_terms_array": size_term_matrix.df.to_numpy(), "timeframe": "trip", + "land_use": state.get_dataframe("land_use"), } ) locals_dict.update(skims)