check py version

huggingface · Jun 25, 2024 · 2c8180a · 2c8180a
1 parent 3bf804f
commit 2c8180a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/transformers/models/donut/processing_donut.py b/src/transformers/models/donut/processing_donut.py
@@ -16,12 +16,13 @@
 Processor class for Donut.
 """
 
+import sys
 from typing import List, Optional, Union
 
 
-try:
+if sys.version_info >= (3, 11):
     from typing import Unpack
-except ImportError:
+else:
     from typing_extensions import Unpack
 
 import re

diff --git a/src/transformers/models/wav2vec2/processing_wav2vec2.py b/src/transformers/models/wav2vec2/processing_wav2vec2.py
@@ -16,13 +16,15 @@
 Speech processor class for Wav2Vec2
 """
 
+import sys
 from typing import List, Optional, Union
 
 
-try:
+if sys.version_info >= (3, 11):
     from typing import Unpack
-except ImportError:
+else:
     from typing_extensions import Unpack
+
 import warnings
 from contextlib import contextmanager