allow numpy to broadcast alt in cdap.apply_final_rules

ActivitySim · May 6, 2015 · 0149190 · 0149190
1 parent bbc6ee0
commit 0149190
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/activitysim/cdap/cdap.py b/activitysim/cdap/cdap.py
@@ -267,9 +267,7 @@ def apply_final_rules(hh_util, people, hh_id_col, final_rules):
             if key in alt_match_cache:
                 alt_match = alt_match_cache[key]
             else:
-                alt_match = (
-                    np.array(utils.index.values.tolist()) ==
-                    np.array([alt] * hh_size))
+                alt_match = np.array(utils.index.values.tolist()) == alt
                 alt_match_cache[key] = alt_match
 
             app = np.any(np.bitwise_and(alt_match, m), axis=1)