lint

Author: Jan Zill

activitysim/abm/models/joint_tour_participation.py

@@ -216,13 +216,17 @@ def participants_chooser(
                     # need "is valid choice" such that we certainly choose those with non-zero values,
                     # and do not choose others. Let's use 3.0 as large value here.
                     probs_or_utils[choice_col] = np.where(
-                        probs_or_utils[choice_col] > logit.UTIL_MIN, 3.0, logit.UTIL_UNAVAILABLE
+                        probs_or_utils[choice_col] > logit.UTIL_MIN,
+                        3.0,
+                        logit.UTIL_UNAVAILABLE,
                     )
                     non_choice_col = [
                         col for col in probs_or_utils.columns if col != choice_col
                     ][0]
                     probs_or_utils[non_choice_col] = np.where(
-                        probs_or_utils[choice_col] <= logit.UTIL_MIN, 3.0, logit.UTIL_UNAVAILABLE
+                        probs_or_utils[choice_col] <= logit.UTIL_MIN,
+                        3.0,
+                        logit.UTIL_UNAVAILABLE,
                     )
                 else:
                     probs_or_utils[choice_col] = np.where(

activitysim/core/interaction_sample.py

@@ -45,8 +45,7 @@ def make_sample_choices_utility_based(
 
     if allow_zero_probs:
         zero_probs = (
-            utilities.sum(axis=1)
-            <= utilities.shape[1] * logit.UTIL_UNAVAILABLE
+            utilities.sum(axis=1) <= utilities.shape[1] * logit.UTIL_UNAVAILABLE
         )
         if zero_probs.all():
             return pd.DataFrame(

activitysim/core/pathbuilder.py

@@ -1013,7 +1013,10 @@ def build_virtual_path(
                     else:
                         if self.network_los.state.settings.use_explicit_error_terms:
                             utilities_df = logit.validate_utils(
-                                self.network_los.state, utilities_df, allow_zero_probs=True, trace_label=trace_label
+                                self.network_los.state,
+                                utilities_df,
+                                allow_zero_probs=True,
+                                trace_label=trace_label,
                             )
                             choices, rands = logit.make_choices_utility_based(
                                 self.network_los.state,
@@ -1043,7 +1046,6 @@ def build_virtual_path(
                         del rands
                         chunk_sizer.log_df(trace_label, "rands", None)
 
-
                 # we need to get path_set, btap, atap from path_df row with same seq and path_num
                 # drop seq join column, but keep path_num of choice to override_choices when tracing
                 columns_to_cache = ["btap", "atap", "path_set", "path_num"]

activitysim/core/simulate.py

@@ -1290,7 +1290,9 @@ def eval_mnl(
         )
 
     if state.settings.use_explicit_error_terms:
-        utilities = logit.validate_utils(state, utilities, trace_label=trace_label, trace_choosers=choosers)
+        utilities = logit.validate_utils(
+            state, utilities, trace_label=trace_label, trace_choosers=choosers
+        )
 
         if custom_chooser:
             choices, rands = custom_chooser(

activitysim/core/test/test_logit.py

@@ -69,8 +69,10 @@ def utilities(choosers, spec, test_data):
         columns=test_data["probabilities"].columns,
     )
 
+
 # TODO-EET: Add tests here!
 
+
 def test_utils_to_probs(utilities, test_data):
     state = workflow.State().default_settings()
     probs = logit.utils_to_probs(state, utilities, trace_label=None)