Fix literal import for python 3.7

materialsproject · Oct 29, 2021 · 0a02a5b · 0a02a5b
1 parent d727115
commit 0a02a5b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/mp_api/routes/_general_store/models.py b/src/mp_api/routes/_general_store/models.py
@@ -1,5 +1,10 @@
 from pydantic import BaseModel, Field
-from typing import Dict, Literal
+from typing import Dict
+
+try:
+    from typing import Literal  # type: ignore
+except ImportError:
+    from typing_extensions import Literal
 from datetime import datetime