Wrap Literal import in try-except ImportError

UKPLab · Dec 12, 2023 · 944cac0 · 944cac0
1 parent 9b9438c
commit 944cac0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sentence_transformers/SentenceTransformer.py b/sentence_transformers/SentenceTransformer.py
@@ -4,7 +4,11 @@
 import shutil
 import stat
 from collections import OrderedDict
-from typing import List, Dict, Tuple, Iterable, Type, Union, Callable, Optional, Literal
+from typing import List, Dict, Tuple, Iterable, Type, Union, Callable, Optional
+try:
+    from typing import Literal
+except ImportError:
+    from typing_extensions import Literal
 import numpy as np
 from numpy import ndarray
 import transformers