fixing disaggregate accessibility bug in zone sampler

SANDAG · Mar 19, 2024 · 627e286 · 627e286
1 parent f969542
commit 627e286
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/activitysim/abm/models/disaggregate_accessibility.py b/activitysim/abm/models/disaggregate_accessibility.py
@@ -275,6 +275,10 @@ 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 = (