Merge 274062c into 494403c

UDST · Dec 7, 2018 · 95bddce · 95bddce
2 parents 494403c + 274062c
commit 95bddce
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/urbansim_templates/models/large_multinomial_logit.py b/urbansim_templates/models/large_multinomial_logit.py
@@ -536,6 +536,9 @@ def probs(mct):
             column = self.out_column
         else:
             column = self.choice_column
+
+        if column not in table.columns:
+            table[column] = None
 
         table.update_col_from_series(column, choices, cast=True)