import Literal from typing_extensions for py37

deepmodeling · Sep 11, 2023 · 6ac8f44 · 6ac8f44
1 parent 08465e4
commit 6ac8f44
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/deepmd/infer/model_devi.py b/deepmd/infer/model_devi.py
@@ -1,6 +1,5 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 from typing import (
-    Literal,
     Optional,
     Tuple,
     overload,
@@ -22,6 +21,11 @@
     DeepPot,
 )
 
+try:
+    from typing import Literal  # python >=3.8
+except ImportError:
+    from typing_extensions import Literal  # type: ignore
+
 
 @overload
 def calc_model_devi_f(

diff --git a/pyproject.toml b/pyproject.toml
@@ -43,7 +43,7 @@ dependencies = [
     'pyyaml',
     'dargs >= 0.3.5',
     'python-hostlist >= 1.21',
-    'typing_extensions; python_version < "3.7"',
+    'typing_extensions; python_version < "3.8"',
     'importlib_metadata>=1.4; python_version < "3.8"',
     'h5py',
     'wcmatch',