suppress import error

ebonnal · Oct 16, 2024 · e5c3475 · e5c3475
1 parent 6b864a6
commit e5c3475
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/streamable/functions.py b/streamable/functions.py
@@ -1,19 +1,22 @@
 import builtins
 import datetime
+from contextlib import suppress
 from typing import (
     Any,
     Callable,
     Coroutine,
     Iterable,
     Iterator,
     List,
-    Literal,
     Optional,
     Type,
     TypeVar,
     cast,
 )
 
+with suppress(ImportError):
+    from typing import Literal
+
 from streamable.iters import (
     AsyncConcurrentMappingIterable,
     ByKeyGroupingIterator,

diff --git a/streamable/iters.py b/streamable/iters.py
@@ -18,7 +18,6 @@
     Iterable,
     Iterator,
     List,
-    Literal,
     NamedTuple,
     Optional,
     Tuple,
@@ -28,6 +27,9 @@
     cast,
 )
 
+with suppress(ImportError):
+    from typing import Literal
+
 from streamable.util.functiontools import catch_and_raise_as
 
 T = TypeVar("T")

diff --git a/streamable/stream.py b/streamable/stream.py
@@ -1,6 +1,6 @@
-from contextlib import suppress
 import datetime
 import logging
+from contextlib import suppress
 from multiprocessing import get_logger
 from typing import (
     TYPE_CHECKING,