update imports

src/maggma/api/utils.py

@@ -3,6 +3,7 @@
 from typing import (
     Any,
     Callable,
+    Literal,
     Optional,
     Union,
     get_args,  # pragma: no cover
@@ -13,7 +14,6 @@
 from pydantic import BaseModel
 from pydantic._internal._utils import lenient_issubclass
 from pydantic.fields import FieldInfo
-from typing_extensions import Literal
 
 from maggma.utils import get_flat_models_from_model
 

src/maggma/stores/mongolike.py

@@ -5,21 +5,10 @@
 """
 
 import warnings
+from collections.abc import Iterator
 from itertools import chain, groupby
 from pathlib import Path
-
-from ruamel.yaml import YAML
-
-from maggma.stores.ssh_tunnel import SSHTunnel
-
-try:
-    from collections.abc import Iterator
-    from typing import Any, Callable, Literal, Optional, Union
-except ImportError:
-    from collections.abc import Iterator
-    from typing import Any, Callable, Optional, Union
-
-    from typing_extensions import Literal
+from typing import Any, Callable, Literal, Optional, Union
 
 import bson
 import mongomock
@@ -31,8 +20,10 @@
 from pydash import get, has, set_
 from pymongo import MongoClient, ReplaceOne, uri_parser
 from pymongo.errors import ConfigurationError, DocumentTooLarge, OperationFailure
+from ruamel.yaml import YAML
 
 from maggma.core import Sort, Store, StoreError
+from maggma.stores.ssh_tunnel import SSHTunnel
 from maggma.utils import confirm_field_index, to_dt
 
 try: