Fixing code quality feedback from CI

RyanMullins · RyanMullins · commit 48ac9e936f24 · 2025-03-20T13:09:26.000Z
diff --git a/tests/models/shieldgemma2/test_processing_shieldgemma2.py b/tests/models/shieldgemma2/test_processing_shieldgemma2.py
@@ -17,7 +17,6 @@
 import os
 import shutil
 import tempfile
-from typing import Optional
 import unittest
 
 from parameterized import parameterized
diff --git a/utils/check_repo.py b/utils/check_repo.py
@@ -167,6 +167,7 @@
     "models/vision_text_dual_encoder/test_modeling_flax_vision_text_dual_encoder.py",
     "models/decision_transformer/test_modeling_decision_transformer.py",
     "models/bark/test_modeling_bark.py",
+    "models/shieldgemma2/test_modeling_shieldgemma2.py",
 ]
 
 # Update this list for models that are not in any of the auto MODEL_XXX_MAPPING. Being in this list is an exception and

Original file line number	Diff line number	Diff line change
`@@ -167,6 +167,7 @@`
`167`	`167`	`"models/vision_text_dual_encoder/test_modeling_flax_vision_text_dual_encoder.py",`
`168`	`168`	`"models/decision_transformer/test_modeling_decision_transformer.py",`
`169`	`169`	`"models/bark/test_modeling_bark.py",`
	`170`	`+ "models/shieldgemma2/test_modeling_shieldgemma2.py",`
`170`	`171`	`]`
`171`	`172`
`172`	`173`	`# Update this list for models that are not in any of the auto MODEL_XXX_MAPPING. Being in this list is an exception and`