remove tazs from sampling if no maz available

SANDAG · Mar 19, 2024 · 1587d6e · 1587d6e
1 parent 94c4db8
commit 1587d6e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/activitysim/abm/models/disaggregate_accessibility.py b/activitysim/abm/models/disaggregate_accessibility.py
@@ -415,6 +415,12 @@ def zone_sampler(self):
                     maz_candidates = maz_candidates[
                         ~maz_candidates.MAZ.isin(maz_sample_idx)
                     ]
+
+                    # Need to make sure we sample from TAZs that still exist in the maz_candidates
+                    taz_candidates = taz_candidates[
+                        taz_candidates.index.isin(maz_candidates.TAZ)
+                    ]
+
                     # Calculate the remaining samples to collect
                     n_samples_remaining = n_samples - len(maz_sample_idx)
                     n_samples_remaining = (