black

Signed-off-by: xadupre <[email protected]>
xadupre · Sep 4, 2024 · 38d2b5b · 38d2b5b
1 parent be13147
commit 38d2b5b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/benchmarks/bench_plot_onnxruntime_linreg.py b/benchmarks/bench_plot_onnxruntime_linreg.py
@@ -130,11 +130,11 @@ def bench(n_obs, n_features, fit_intercepts, methods, repeat=10, verbose=False):
 
                     # checks that both produce the same outputs
                     if n <= 10000 and len(p1.shape) == 1 and len(p2.shape) == 2:
-                            p2 = p2.ravel()
-                            try:
-                                assert_almost_equal(p1.ravel(), p2.ravel(), decimal=5)
-                            except AssertionError as e:
-                                warnings.warning(str(e))
+                        p2 = p2.ravel()
+                        try:
+                            assert_almost_equal(p1.ravel(), p2.ravel(), decimal=5)
+                        except AssertionError as e:
+                            warnings.warning(str(e))
     return res