Previous commit is only necessary for 2- and 3-zone systems

SANDAG · Feb 21, 2024 · 5fb2b83 · 5fb2b83
1 parent 851319d
commit 5fb2b83
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/activitysim/abm/models/trip_matrices.py b/activitysim/abm/models/trip_matrices.py
@@ -246,12 +246,13 @@ def write_trip_matrices(network_los):
         trips_df["destination"] = trips_df["true_destination"]
         del trips_df["true_origin"], trips_df["true_destination"]
 
-        trips_df["otaz"] = (
-            pipeline.get_table("land_use").reindex(trips_df["origin"]).TAZ.tolist()
-        )
-        trips_df["dtaz"] = (
-            pipeline.get_table("land_use").reindex(trips_df["destination"]).TAZ.tolist()
-        )
+        if network_los.zone_system == los.TWO_ZONE or network_los.zone_system == los.THREE_ZONE:
+            trips_df["otaz"] = (
+                pipeline.get_table("land_use").reindex(trips_df["origin"]).TAZ.tolist()
+            )
+            trips_df["dtaz"] = (
+                pipeline.get_table("land_use").reindex(trips_df["destination"]).TAZ.tolist()
+            )
 
 
 def annotate_trips(trips, network_los, model_settings):