reformat

hoshibara · hoshibara · commit 58a681a9deb8 · 2025-03-13T07:29:28.000Z
diff --git a/test/inductor/test_flex_attention.py b/test/inductor/test_flex_attention.py
@@ -3262,7 +3262,7 @@ def test_mixed_device_error_message(self):
     @supported_platform
     def test_captured_wrong_device_error_message(self):
         means = torch.randn(64, 3, device=self.device)
-        length_scales = torch.logspace(0.001, 0.1, 8, device='cpu')
+        length_scales = torch.logspace(0.001, 0.1, 8, device="cpu")
 
         def euclidean_dist_pos_embed(score, b, h, q_idx, k_idx):
             q_pos = means[q_idx]
