nan prob fixes

LBNL-UCB-STI · Jul 29, 2024 · 53aa11d · 53aa11d
1 parent 2bd026e
commit 53aa11d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/activitysim/core/interaction_sample_simulate.py b/activitysim/core/interaction_sample_simulate.py
@@ -390,7 +390,7 @@ def interaction_sample_simulate(
     alternatives,
     spec,
     choice_column,
-    allow_zero_probs=False,
+    allow_zero_probs=True,
     zero_prob_choice_val=None,
     log_alt_losers=False,
     want_logsums=False,

diff --git a/activitysim/core/logit.py b/activitysim/core/logit.py
@@ -327,8 +327,8 @@ def make_choices(
             trace_choosers=trace_choosers,
             raise_error=False
         )
-        bad_probs[bad_probs, :] = 0
-        bad_probs[bad_probs, 0] = 1
+        bad_probs.loc[bad_probs, :] = 0
+        bad_probs.loc[bad_probs, 0] = 1
 
     rands = state.get_rn_generator().random_for_df(probs)