From 540a10848c26ebec9a0e749d3808333bdae08167 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 27 Jun 2025 12:28:10 +0200 Subject: [PATCH] fix `Gemma3nProcessorTest` (#39068) * fix * fix * oups forgot style --------- Co-authored-by: ydshieh Co-authored-by: Cyril Vallez --- tests/models/gemma3n/test_processing_gemma3n.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/models/gemma3n/test_processing_gemma3n.py b/tests/models/gemma3n/test_processing_gemma3n.py index 1d30a80c48..ffedb2b98a 100644 --- a/tests/models/gemma3n/test_processing_gemma3n.py +++ b/tests/models/gemma3n/test_processing_gemma3n.py @@ -36,7 +36,7 @@ if is_speech_available(): class Gemma3nProcessorTest(unittest.TestCase): def setUp(self): # TODO: update to google? - self.model_id = "Google/gemma-3n-E4B-it" + self.model_id = "hf-internal-testing/namespace-google-repo_name-gemma-3n-E4B-it" self.tmpdirname = tempfile.mkdtemp(suffix="gemma3n") self.maxDiff = None @@ -71,6 +71,9 @@ class Gemma3nProcessorTest(unittest.TestCase): self.assertIsInstance(processor.tokenizer, GemmaTokenizerFast) self.assertEqual(processor.tokenizer.get_vocab(), tokenizer.get_vocab()) + # `disable_grouping` is a new attribute that got added on main while gemma3n was being released - so was + # not part of the saved processor + del processor.feature_extractor.disable_grouping self.assertIsInstance(processor.feature_extractor, Gemma3nAudioFeatureExtractor) self.assertEqual(processor.feature_extractor.to_json_string(), feature_extractor.to_json_string()) @@ -94,6 +97,9 @@ class Gemma3nProcessorTest(unittest.TestCase): self.assertEqual(processor.tokenizer.get_vocab(), tokenizer_add_kwargs.get_vocab()) self.assertIsInstance(processor.tokenizer, GemmaTokenizerFast) + # `disable_grouping` is a new attribute that got added on main while gemma3n was being released - so was + # not part of the saved processor + del processor.feature_extractor.disable_grouping self.assertEqual(processor.feature_extractor.to_json_string(), feature_extractor_add_kwargs.to_json_string()) self.assertIsInstance(processor.feature_extractor, Gemma3nAudioFeatureExtractor)