style

huggingface · Dec 13, 2023 · 3a8e079 · 3a8e079
1 parent 2fd4105
commit 3a8e079
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/vit_msn/test_modeling_vit_msn.py b/tests/models/vit_msn/test_modeling_vit_msn.py
@@ -227,6 +227,6 @@ def test_inference_image_classification_head(self):
         expected_shape = torch.Size((1, 1000))
         self.assertEqual(outputs.logits.shape, expected_shape)
 
-        expected_slice = torch.tensor([ 0.5588,  0.6853, -0.5929]).to(torch_device)
+        expected_slice = torch.tensor([0.5588, 0.6853, -0.5929]).to(torch_device)
 
         self.assertTrue(torch.allclose(outputs.logits[0, :3], expected_slice, atol=1e-4))