adding the version check for transformers

OpenNMT · Sep 11, 2024 · b3b8089 · b3b8089
1 parent 7a1be4d
commit b3b8089
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/tests/test_transformers.py b/python/tests/test_transformers.py
@@ -1054,6 +1054,11 @@ def test_transformers_wav2vec2bert(
         import torch
         import transformers
 
+        required_version = "4.41.0"
+        current_version = transformers.__version__
+        if current_version < required_version:
+            pytest.skip(f"Transformers version must be >= {required_version}, but found {current_version}")
+
         converter = ctranslate2.converters.TransformersConverter(
             model_name, load_as_float16="int8"
         )