Black

NKI-AI · Jun 27, 2024 · 1be532a · 1be532a
1 parent d05224e
commit 1be532a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/tests_utils/test_utils.py b/tests/tests_utils/test_utils.py
@@ -138,8 +138,8 @@ def test_normalize_image(shape, eps):
         (np.random.rand(2, 3), (2, 1, 3), (2, 1, 3)),
         (np.random.rand(4), (1, 1, 4, 1), (1, 1, 4, 1)),
         (np.random.rand(6), (1, 6, 1), (1, 6, 1)),
-    ]
+    ],
 )
 def test_reshape_array_to_shape(array, requested_shape, expected_shape):
     result = reshape_array_to_shape(array, requested_shape)
-    assert result.shape == expected_shape
+    assert result.shape == expected_shape