linting fixes on multiarea run method

Transport-for-the-North · Jan 8, 2025 · 5cee9b7 · 5cee9b7
1 parent 9997ce0
commit 5cee9b7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/caf/distribute/gravity_model/multi_area.py b/src/caf/distribute/gravity_model/multi_area.py
@@ -921,6 +921,7 @@ def run(
                 cost_distribution=self.achieved_cost_dist[i],
                 cost_convergence=self.achieved_convergence[dist.name],
                 value_distribution=self.achieved_distribution[dist.zones],
+                trafe_cost_distribution=dist.cost_distribution,
                 cost_function=self.cost_function,
                 cost_params=self._cost_params_to_kwargs(
                     cost_args[i * params_len : i * params_len + params_len]