fix: typing imports

darts/explainability/tft_explainer.py

@@ -23,7 +23,7 @@
 """
 
 from collections.abc import Sequence
-from typing import Optional, Union
+from typing import Literal, Optional, Union
 
 import matplotlib.axes
 import matplotlib.pyplot as plt
@@ -38,12 +38,6 @@
 from darts.models import TFTModel
 from darts.utils.utils import generate_index
 
-try:
-    from typing import Literal
-except ImportError:
-    from typing_extensions import Literal
-
-
 logger = get_logger(__name__)
 
 

darts/models/forecasting/arima.py

@@ -10,12 +10,13 @@
 .. [1] https://wikipedia.org/wiki/Autoregressive_integrated_moving_average
 """
 
+import sys
 from collections.abc import Sequence
 from typing import Literal, Optional, Union
 
-try:
+if sys.version_info >= (3, 10):
     from typing import TypeAlias
-except ImportError:
+else:
     from typing_extensions import TypeAlias
 
 import numpy as np

darts/timeseries.py

@@ -42,7 +42,7 @@
 from copy import deepcopy
 from inspect import signature
 from io import StringIO
-from typing import Any, Callable, Optional, Union
+from typing import Any, Callable, Literal, Optional, Union
 
 import matplotlib.axes
 import matplotlib.pyplot as plt
@@ -60,11 +60,6 @@
     n_steps_between,
 )
 
-try:
-    from typing import Literal
-except ImportError:
-    from typing_extensions import Literal
-
 if sys.version_info >= (3, 11):
     from typing import Self
 else: