simple formatting

CyrilJl · Jan 2, 2025 · 46fe08c · 46fe08c
1 parent 9815259
commit 46fe08c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/optimask/_optimask.py b/optimask/_optimask.py
@@ -353,4 +353,4 @@ def solve(
             if return_data:
                 return X[np.ix_(rows, cols)].copy()
             else:
-                return rows, cols
+                return rows, cols
diff --git a/tests/test_optimask.py b/tests/test_optimask.py
@@ -54,12 +54,12 @@ def test_one_col(opti_mask_instance):
     x = np.arange(m, dtype=float)[:, None]
     x[-1, 0] = np.nan
 
-    assert np.allclose(np.arange(m-1)[:, None], opti_mask_instance.solve(x, return_data=True, check_result=True))
+    assert np.allclose(np.arange(m - 1)[:, None], opti_mask_instance.solve(x, return_data=True, check_result=True))
 
 
 def test_one_row(opti_mask_instance):
     m = 100
     x = np.arange(m, dtype=float)[None, :]
     x[0, -1] = np.nan
 
-    assert np.allclose(np.arange(m-1)[None, :], opti_mask_instance.solve(x, return_data=True, check_result=True))
+    assert np.allclose(np.arange(m - 1)[None, :], opti_mask_instance.solve(x, return_data=True, check_result=True))