minor

huggingface · May 3, 2024 · bff7fc3 · bff7fc3
1 parent 62f8830
commit bff7fc3
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/models/vision_text_dual_encoder/test_modeling_vision_text_dual_encoder.py b/tests/models/vision_text_dual_encoder/test_modeling_vision_text_dual_encoder.py
@@ -21,7 +21,14 @@
 
 import numpy as np
 
-from transformers.testing_utils import is_pt_flax_cross_test, require_torch, require_vision, slow, torch_device
+from transformers.testing_utils import (
+    _run_slow_tests,
+    is_pt_flax_cross_test,
+    require_torch,
+    require_vision,
+    slow,
+    torch_device,
+)
 from transformers.utils import is_flax_available, is_torch_available, is_vision_available
 
 from ...test_modeling_common import floats_tensor, ids_tensor, random_attention_mask