Ruff didn't like the default value for include_leaf_inequalities

cog-imperial · Oct 3, 2024 · 0231f0e · 0231f0e
1 parent e7def05
commit 0231f0e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/omlt/linear_tree/lt_formulation.py b/src/omlt/linear_tree/lt_formulation.py
@@ -106,6 +106,7 @@ def _build_formulation(self):
             output_vars=self.block.scaled_outputs,
             transformation=self.transformation,
             epsilon=self.epsilon,
+            include_leaf_equalities=True
         )
 
 
@@ -266,7 +267,7 @@ def _add_gdp_formulation_to_block(  # noqa: PLR0913
     output_vars,
     transformation,
     epsilon,
-    include_leaf_equalities=True,
+    include_leaf_equalities,
 ):
     """This function adds the GDP representation to the OmltBlock using Pyomo.GDP.