From 26be7f717e12241df3d575d825bfa51026da9be8 Mon Sep 17 00:00:00 2001 From: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com> Date: Wed, 16 Jul 2025 15:53:43 +0200 Subject: [PATCH] [`CI`] Fix partially red CI (#39448) fix --- tests/models/git/test_modeling_git.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/models/git/test_modeling_git.py b/tests/models/git/test_modeling_git.py index 36683777b4..38aa2b4e87 100644 --- a/tests/models/git/test_modeling_git.py +++ b/tests/models/git/test_modeling_git.py @@ -347,8 +347,7 @@ class GitModelTester: num_return_sequences=2, ) - self.parent.assertEqual(generated_ids.shape[0], self.batch_size * 2) - self.parent.assertTrue(generated_ids.shape[1] < 20) + self.parent.assertEqual(generated_ids.shape, (self.batch_size * 2, 20)) def _test_batched_generate_captioning(self, config, input_ids, input_mask, pixel_values): model = GitForCausalLM(config=config)