SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Apply several ruff SIM rules (#37283)

Browse Source 
* Apply ruff SIM118 fix

Signed-off-by: cyy <cyyever@outlook.com>

* Apply ruff SIM910 fix

Signed-off-by: cyy <cyyever@outlook.com>

* Apply ruff SIM101 fix

Signed-off-by: cyy <cyyever@outlook.com>

* Format code

Signed-off-by: cyy <cyyever@outlook.com>

* More fixes

Signed-off-by: cyy <cyyever@outlook.com>

---------

Signed-off-by: cyy <cyyever@outlook.com>
This commit is contained in:
Yuanyuan Chen
2025-07-29 19:40:34 +08:00
committed by GitHub
parent cf97f6cfd1
commit 95faabf0a6
391 changed files with 762 additions and 788 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/gemma3n/test_processing_gemma3n.py
View File

@@ -116,7 +116,7 @@ class Gemma3nProcessorTest(unittest.TestCase):
input_image_processor = image_processor(raw_image, return_tensors="pt")
input_processor = processor(text="Describe:", images=raw_image, return_tensors="pt")
for key in input_image_processor.keys():
for key in input_image_processor:
self.assertAlmostEqual(input_image_processor[key].sum(), input_processor[key].sum(), delta=1e-2)
if "pixel_values" in key:
# NOTE: all images should be re-scaled to 768x768
@@ -135,7 +135,7 @@ class Gemma3nProcessorTest(unittest.TestCase):
input_feat_extract = feature_extractor(raw_speech, return_tensors="pt")
input_processor = processor(text="Transcribe:", audio=raw_speech, return_tensors="pt")
for key in input_feat_extract.keys():
for key in input_feat_extract:
self.assertAlmostEqual(input_feat_extract[key].sum(), input_processor[key].sum(), delta=1e-2)
def test_tokenizer(self):
@@ -152,7 +152,7 @@ class Gemma3nProcessorTest(unittest.TestCase):
encoded_tok = tokenizer(input_str)
for key in encoded_tok.keys():
for key in encoded_tok:
self.assertListEqual(encoded_tok[key], encoded_processor[key][0])
def test_tokenizer_decode(self):