formatted files

keras-team · Oct 7, 2024 · e12950c · e12950c
1 parent 738375f
commit e12950c
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/keras/src/ops/nn_test.py b/keras/src/ops/nn_test.py
@@ -2235,9 +2235,15 @@ def test_dot_product_attention(
                 mask_shape
             )
 
-        if flash_attention and backend.backend() in ["torch", "tensorflow", "numpy"]:
-            self.skipTest("Not supported in TF and NumPy and supported for "
-                          "PyTorch with specific requirements.")
+        if flash_attention and backend.backend() in [
+            "torch",
+            "tensorflow",
+            "numpy",
+        ]:
+            self.skipTest(
+                "Not supported in TF and NumPy and supported for "
+                "PyTorch with specific requirements."
+            )
 
         expected = _dot_product_attention(
             query,