black

Pyomo · Aug 13, 2024 · 10b9026 · 10b9026
1 parent e890a25
commit 10b9026
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/pyomo/contrib/piecewise/tests/test_nonlinear_to_pwl.py b/pyomo/contrib/piecewise/tests/test_nonlinear_to_pwl.py
@@ -371,7 +371,7 @@ def test_uniform_sampling_discrete_vars(self):
         for x in [0, 1]:
             for y in [0, 1]:
                 for z in [0, 2, 5]:
-                    self.assertIn((x, y, z), points)    
+                    self.assertIn((x, y, z), points)
 
     @unittest.skipUnless(numpy_available, "Numpy is not available")
     def test_random_sampling_discrete_vars(self):

diff --git a/pyomo/contrib/piecewise/transform/nonlinear_to_pwl.py b/pyomo/contrib/piecewise/transform/nonlinear_to_pwl.py
@@ -99,8 +99,9 @@ def _get_random_point_grid(bounds, n, func, config, seed=42):
             linspaces.append(np.random.uniform(lb, ub, n))
         else:
             size = min(n, ub - lb + 1)
-            linspaces.append(np.random.choice(range(lb, ub + 1), size=size,
-                                              replace=False))
+            linspaces.append(
+                np.random.choice(range(lb, ub + 1), size=size, replace=False)
+            )
     return list(itertools.product(*linspaces))
 
 
@@ -169,7 +170,7 @@ def _get_pwl_function_approximation(func, config, bounds):
     func: function to approximate
     config: ConfigDict for transformation, specifying domain_partitioning_method,
        num_points, and max_depth (if using linear trees)
-    bounds: list of tuples giving upper and lower bounds and a boolean indicating 
+    bounds: list of tuples giving upper and lower bounds and a boolean indicating
        if the variable's domain is discrete or not, for each of func's arguments
     """
     method = config.domain_partitioning_method