fix black

awslabs · Dec 14, 2022 · 2fc8c2e · 2fc8c2e
1 parent 14b84b6
commit 2fc8c2e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/torch/model/test_modules.py b/test/torch/model/test_modules.py
@@ -64,7 +64,11 @@ def assert_shapes_and_dtypes(tensors, shapes, dtypes):
             ),
             4,
             [[(4, 12), (4, 12), (4, 12)], (4, 1), (4, 1)],
-            [[torch.float, torch.float, torch.float], torch.float, torch.float],
+            [
+                [torch.float, torch.float, torch.float],
+                torch.float,
+                torch.float,
+            ],
         ),
     ],
 )