Fix: Addressed 'max' test failures in pull request

huggingface · Dec 13, 2024 · 9919701 · 9919701
1 parent eb2f730
commit 9919701
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modeling_flax_common.py b/tests/test_modeling_flax_common.py
@@ -24,7 +24,7 @@
 import transformers
 from transformers import is_flax_available, is_torch_available
 from transformers.models.auto import get_values
-from transformers.testing_utils import CaptureLogger, is_pt_flax_cross_test, require_flax, require_torch, torch_device
+from transformers.testing_utils import CaptureLogger, is_pt_flax_cross_test, require_flax, torch_device
 from transformers.utils import CONFIG_NAME, GENERATION_CONFIG_NAME, logging
 from transformers.utils.generic import ModelOutput