format

uxlfoundation · Nov 3, 2024 · 8659248 · 8659248
1 parent 2f85fd4
commit 8659248
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/sklearnex/tests/test_memory_usage.py b/sklearnex/tests/test_memory_usage.py
@@ -142,14 +142,14 @@ class DummyEstimatorWithTableConversions(BaseEstimator):
 
         def fit(self, X, y=None):
             sua_iface, xp, _ = _get_sycl_namespace(X)
-            assert X.flags['C_CONTIGUOUS'] or X.flags['F_CONTIGUOUS']
-            assert y.flags['C_CONTIGUOUS'] or y.flags['F_CONTIGUOUS']
-            if not (X.flags['C_CONTIGUOUS'] or X.flags['F_CONTIGUOUS']):
+            assert X.flags["C_CONTIGUOUS"] or X.flags["F_CONTIGUOUS"]
+            assert y.flags["C_CONTIGUOUS"] or y.flags["F_CONTIGUOUS"]
+            if not (X.flags["C_CONTIGUOUS"] or X.flags["F_CONTIGUOUS"]):
                 X = xp.copy(X)
-            if not (y.flags['C_CONTIGUOUS'] or y.flags['F_CONTIGUOUS']):
+            if not (y.flags["C_CONTIGUOUS"] or y.flags["F_CONTIGUOUS"]):
                 y = xp.copy(y)
-            assert X.flags['C_CONTIGUOUS'] or X.flags['F_CONTIGUOUS']
-            assert y.flags['C_CONTIGUOUS'] or y.flags['F_CONTIGUOUS']
+            assert X.flags["C_CONTIGUOUS"] or X.flags["F_CONTIGUOUS"]
+            assert y.flags["C_CONTIGUOUS"] or y.flags["F_CONTIGUOUS"]
             X_table = to_table(X)
             y_table = to_table(y)
             # The presence of the fitted attributes (ending with a trailing
@@ -168,10 +168,10 @@ def predict(self, X):
             # fitted attributes (ending with a trailing underscore).
             check_is_fitted(self)
             sua_iface, xp, _ = _get_sycl_namespace(X)
-            assert X.flags['C_CONTIGUOUS'] or X.flags['F_CONTIGUOUS']
-            if not (X.flags['C_CONTIGUOUS'] or X.flags['F_CONTIGUOUS']):
+            assert X.flags["C_CONTIGUOUS"] or X.flags["F_CONTIGUOUS"]
+            if not (X.flags["C_CONTIGUOUS"] or X.flags["F_CONTIGUOUS"]):
                 X = xp.copy(X)
-            assert X.flags['C_CONTIGUOUS'] or X.flags['F_CONTIGUOUS']
+            assert X.flags["C_CONTIGUOUS"] or X.flags["F_CONTIGUOUS"]
             X_table = to_table(X)
             returned_X = from_table(
                 X_table, sua_iface=sua_iface, sycl_queue=X.sycl_queue, xp=xp