ensure mct in large mnl is filtered by segmentation column for segmen…

…ted large mnl use case
UDST · Apr 13, 2020 · 6d6f938 · 6d6f938
1 parent 0b55c25
commit 6d6f938
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/urbansim_templates/models/large_multinomial_logit.py b/urbansim_templates/models/large_multinomial_logit.py
@@ -463,8 +463,9 @@ def fit(self, mct=None):
             df_from_mct = mct.to_frame()
             idx_names = df_from_mct.index.names
             df_from_mct = df_from_mct.reset_index()
-            df_from_mct = apply_filter_query(df_from_mct, self.chooser_filters)
-            mct = MergedChoiceTable.from_df(df_from_mct).set_index(idx_names)
+            df_from_mct = apply_filter_query(
+                df_from_mct, self.chooser_filters).set_index(idx_names)
+            mct = MergedChoiceTable.from_df(df_from_mct)
 
         else:        
             observations = get_data(tables = self.choosers,