anemoi-datasets 0.5.25__py3-none-any.whl → 0.5.27__py3-none-any.whl

anemoi/datasets/create/input/result.py

@@ -15,10 +15,6 @@ from collections import defaultdict
 from functools import cached_property
 from typing import Any
 from typing import DefaultDict
-from typing import Dict
-from typing import List
-from typing import Optional
-from typing import Tuple
 
 import numpy as np
 from anemoi.utils.dates import as_timedelta
@@ -33,7 +29,7 @@ from .trace import trace_datasource
 LOG = logging.getLogger(__name__)
 
 
-def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
+def _fields_metatata(variables: tuple[str, ...], cube: Any) -> dict[str, Any]:
     """Retrieve metadata for the given variables and cube.
 
     Parameters
@@ -50,7 +46,7 @@ def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
     """
     assert isinstance(variables, tuple), variables
 
-    KNOWN: Dict[str, Dict[str, bool]] = {
+    KNOWN: dict[str, dict[str, bool]] = {
         "cos_julian_day": dict(computed_forcing=True, constant_in_time=False),
         "cos_latitude": dict(computed_forcing=True, constant_in_time=True),
         "cos_local_time": dict(computed_forcing=True, constant_in_time=False),
@@ -65,9 +61,9 @@ def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
         "sin_longitude": dict(computed_forcing=True, constant_in_time=True),
     }
 
-    def _merge(md1: Dict[str, Any], md2: Dict[str, Any]) -> Dict[str, Any]:
+    def _merge(md1: dict[str, Any], md2: dict[str, Any]) -> dict[str, Any]:
         assert set(md1.keys()) == set(md2.keys()), (set(md1.keys()), set(md2.keys()))
-        result: Dict[str, Any] = {}
+        result: dict[str, Any] = {}
         for k in md1.keys():
             v1 = md1[k]
             v2 = md2[k]
@@ -90,10 +86,10 @@ def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
 
         return result
 
-    mars: Dict[str, Any] = {}
-    other: DefaultDict[str, Dict[str, Any]] = defaultdict(dict)
+    mars: dict[str, Any] = {}
+    other: DefaultDict[str, dict[str, Any]] = defaultdict(dict)
     i: int = -1
-    date: Optional[str] = None
+    date: str | None = None
     for c in cube.iterate_cubelets():
 
         if date is None:
@@ -137,7 +133,7 @@ def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
 
         if startStep != endStep:
             # https://codes.ecmwf.int/grib/format/grib2/ctables/4/10/
-            TYPE_OF_STATISTICAL_PROCESSING: Dict[Optional[int], Optional[str]] = {
+            TYPE_OF_STATISTICAL_PROCESSING: dict[int | None, str | None] = {
                 None: None,
                 0: "average",
                 1: "accumulation",
@@ -157,12 +153,12 @@ def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
 
             # https://codes.ecmwf.int/grib/format/grib1/ctable/5/
 
-            TIME_RANGE_INDICATOR: Dict[int, str] = {
+            TIME_RANGE_INDICATOR: dict[int, str] = {
                 4: "accumulation",
                 3: "average",
             }
 
-            STEP_TYPE_FOR_CONVERSION: Dict[str, str] = {
+            STEP_TYPE_FOR_CONVERSION: dict[str, str] = {
                 "min": "minimum",
                 "max": "maximum",
                 "accum": "accumulation",
@@ -172,7 +168,7 @@ def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
             # A few patches
             #
 
-            PATCHES: Dict[str, str] = {
+            PATCHES: dict[str, str] = {
                 "10fg6": "maximum",
                 "mntpr3": "minimum",  # Not in param db
                 "mntpr6": "minimum",  # Not in param db
@@ -211,7 +207,7 @@ def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
         else:
             mars[variables[i]] = md
 
-    result: Dict[str, Dict[str, Any]] = {}
+    result: dict[str, dict[str, Any]] = {}
     for k, v in mars.items():
         result[k] = dict(mars=v) if v else {}
         result[k].update(other[k])
@@ -222,7 +218,7 @@ def _fields_metatata(variables: Tuple[str, ...], cube: Any) -> Dict[str, Any]:
     return result
 
 
-def _data_request(data: Any) -> Dict[str, Any]:
+def _data_request(data: Any) -> dict[str, Any]:
     """Build a data request dictionary from the given data.
 
     Parameters
@@ -235,12 +231,12 @@ def _data_request(data: Any) -> Dict[str, Any]:
     dict
         The data request dictionary.
     """
-    date: Optional[Any] = None
+    date: Any | None = None
     params_levels: DefaultDict[str, set] = defaultdict(set)
     params_steps: DefaultDict[str, set] = defaultdict(set)
 
-    area: Optional[Any] = None
-    grid: Optional[Any] = None
+    area: Any | None = None
+    grid: Any | None = None
 
     for field in data:
         try:
@@ -270,8 +266,8 @@ def _data_request(data: Any) -> Dict[str, Any]:
         except Exception:
             LOG.error(f"Error in retrieving metadata (cannot build data request info) for {field}", exc_info=True)
 
-    def sort(old_dic: DefaultDict[str, set]) -> Dict[str, List[Any]]:
-        new_dic: Dict[str, List[Any]] = {}
+    def sort(old_dic: DefaultDict[str, set]) -> dict[str, list[Any]]:
+        new_dic: dict[str, list[Any]] = {}
         for k, v in old_dic.items():
             new_dic[k] = sorted(list(v))
         return new_dic
@@ -288,7 +284,7 @@ class Result:
     empty: bool = False
     _coords_already_built: bool = False
 
-    def __init__(self, context: ActionContext, action_path: List[str], dates: Any) -> None:
+    def __init__(self, context: ActionContext, action_path: list[str], dates: Any) -> None:
         """Initialize a Result instance.
 
         Parameters
@@ -309,7 +305,7 @@ class Result:
 
         self.context: Any = context
         self.group_of_dates: Any = dates
-        self.action_path: List[str] = action_path
+        self.action_path: list[str] = action_path
 
     @property
     @trace_datasource
@@ -318,7 +314,7 @@ class Result:
         self._raise_not_implemented()
 
     @property
-    def data_request(self) -> Dict[str, Any]:
+    def data_request(self) -> dict[str, Any]:
         """Returns a dictionary with the parameters needed to retrieve the data."""
         return _data_request(self.datasource)
 
@@ -340,7 +336,7 @@ class Result:
         LOG.debug("Sorting dataset %s %s", dict(order_by), remapping)
         assert order_by, order_by
 
-        patches: Dict[str, Dict[Optional[Any], int]] = {"number": {None: 0}}
+        patches: dict[str, dict[Any | None, int]] = {"number": {None: 0}}
 
         try:
             cube: Any = ds.cube(
@@ -377,7 +373,7 @@ class Result:
         patches : Any
             The patches configuration.
         """
-        METADATA: Tuple[str, ...] = (
+        METADATA: tuple[str, ...] = (
             "date",
             "time",
             "step",
@@ -402,7 +398,7 @@ class Result:
         # print("Executing", self.action_path)
         # print("Dates:", compress_dates(self.dates))
 
-        names: List[str] = []
+        names: list[str] = []
         for a in args:
             if isinstance(a, str):
                 names.append(a)
@@ -421,7 +417,7 @@ class Result:
                 print("     ", n)
 
         print()
-        user_shape: Tuple[int, ...] = tuple(len(v) for k, v in user_coords.items())
+        user_shape: tuple[int, ...] = tuple(len(v) for k, v in user_coords.items())
         print("Shape of the hypercube           :", user_shape)
         print(
             "Number of expected fields        :", math.prod(user_shape), "=", " x ".join([str(i) for i in user_shape])
@@ -643,13 +639,13 @@ class Result:
         self._coords_already_built: bool = True
 
     @property
-    def variables(self) -> List[str]:
+    def variables(self) -> list[str]:
         """Retrieve the variables for the result."""
         self.build_coords()
         return self._variables
 
     @property
-    def variables_metadata(self) -> Dict[str, Any]:
+    def variables_metadata(self) -> dict[str, Any]:
         """Retrieve the metadata for the variables."""
         return _fields_metatata(self.variables, self._cube)
 
@@ -690,7 +686,7 @@ class Result:
         return self._proj_string
 
     @cached_property
-    def shape(self) -> List[int]:
+    def shape(self) -> list[int]:
         """Retrieve the shape of the result."""
         return [
             len(self.group_of_dates),
@@ -700,7 +696,7 @@ class Result:
         ]
 
     @cached_property
-    def coords(self) -> Dict[str, Any]:
+    def coords(self) -> dict[str, Any]:
         """Retrieve the coordinates of the result."""
         return {
             "dates": list(self.group_of_dates),

anemoi/datasets/create/input/step.py

@@ -8,13 +8,8 @@
 # nor does it submit to any jurisdiction.
 
 import logging
-import warnings
 from copy import deepcopy
 from typing import Any
-from typing import Dict
-from typing import List
-from typing import Optional
-from typing import Type
 
 from .action import Action
 from .action import ActionContext
@@ -31,7 +26,7 @@ class StepResult(Result):
     """Represents the result of a step in the data processing pipeline."""
 
     def __init__(
-        self, context: Context, action_path: List[str], group_of_dates: Any, action: Action, upstream_result: Result
+        self, context: Context, action_path: list[str], group_of_dates: Any, action: Action, upstream_result: Result
     ) -> None:
         """Initialize a StepResult instance.
 
@@ -64,10 +59,10 @@ class StepResult(Result):
 class StepAction(Action):
     """Represents an action that is part of a step in the data processing pipeline."""
 
-    result_class: Optional[Type[StepResult]] = None
+    result_class: type[StepResult] | None = None
 
     def __init__(
-        self, context: ActionContext, action_path: List[str], previous_step: Any, *args: Any, **kwargs: Any
+        self, context: ActionContext, action_path: list[str], previous_step: Any, *args: Any, **kwargs: Any
     ) -> None:
         """Initialize a StepAction instance.
 
@@ -116,7 +111,7 @@ class StepAction(Action):
         return self._repr(self.previous_step, _inline_=str(self.kwargs))
 
 
-def step_factory(config: Dict[str, Any], context: ActionContext, action_path: List[str], previous_step: Any) -> Any:
+def step_factory(config: dict[str, Any], context: ActionContext, action_path: list[str], previous_step: Any) -> Any:
     """Factory function to create a step action based on the given configuration.
 
     Parameters
@@ -165,24 +160,11 @@ def step_factory(config: Dict[str, Any], context: ActionContext, action_path: Li
     if cls is not None:
         return cls(context, action_path, previous_step, *args, **kwargs)
 
-    # Try filters from datasets filter registry
+    # Try filters from transform filter registry
     from anemoi.transform.filters import filter_registry as transform_filter_registry
 
-    from ..filters import create_filter as create_datasets_filter
-    from ..filters import filter_registry as datasets_filter_registry
-
-    if datasets_filter_registry.is_registered(key):
-
-        if transform_filter_registry.is_registered(key):
-            warnings.warn(f"Filter `{key}` is registered in both datasets and transform filter registries")
-
-        filter = create_datasets_filter(None, config)
-        return FunctionStepAction(context, action_path + [key], previous_step, key, filter)
-
-    # Use filters from transform registry
-
     if transform_filter_registry.is_registered(key):
-        from ..filters.transform import TransformFilter
+        from ..filter import TransformFilter
 
         return FunctionStepAction(
             context, action_path + [key], previous_step, key, TransformFilter(context, key, config)

anemoi/datasets/create/input/template.py

@@ -12,10 +12,9 @@ import logging
 import re
 from abc import ABC
 from abc import abstractmethod
+from collections.abc import Callable
 from functools import wraps
 from typing import Any
-from typing import Callable
-from typing import List
 
 from .context import Context
 
@@ -68,7 +67,7 @@ class Substitution(ABC):
 class Reference(Substitution):
     """A class to represent a reference to another value in the context."""
 
-    def __init__(self, context: Any, action_path: List[str]) -> None:
+    def __init__(self, context: Any, action_path: list[str]) -> None:
         """Initialize a Reference instance.
 
         Parameters
@@ -79,7 +78,7 @@ class Reference(Substitution):
             The action path to resolve.
         """
         self.context: Any = context
-        self.action_path: List[str] = action_path
+        self.action_path: list[str] = action_path
 
     def resolve(self, context: Context) -> Any:
         """Resolve the reference using the given context.

anemoi/datasets/create/input/trace.py

@@ -11,9 +11,9 @@
 import logging
 import textwrap
 import threading
+from collections.abc import Callable
 from functools import wraps
 from typing import Any
-from typing import Callable
 
 LOG = logging.getLogger(__name__)
 

anemoi/datasets/create/patch.py

@@ -10,14 +10,13 @@
 import json
 import logging
 import os
-from typing import Union
 
 import zarr
 
 LOG = logging.getLogger(__name__)
 
 
-def fix_order_by(order_by: Union[dict, list]) -> list[dict]:
+def fix_order_by(order_by: dict | list) -> list[dict]:
     """Fix the order_by attribute to ensure it is a list of dictionaries.
 
     Parameters

anemoi/datasets/create/persistent.py

@@ -16,9 +16,8 @@ import os
 import pickle
 import shutil
 import socket
+from collections.abc import Iterator
 from typing import Any
-from typing import Iterator
-from typing import Tuple
 
 import numpy as np
 from anemoi.utils.provenance import gather_provenance_info
@@ -198,7 +197,7 @@ class BufferedPersistentDict(PersistentDict):
         self.elements = []
         self.keys = []
 
-    def items(self) -> Iterator[Tuple[Any, Any]]:
+    def items(self) -> Iterator[tuple[Any, Any]]:
         """Yield items stored in the BufferedPersistentDict.
 
         Yields
@@ -207,8 +206,7 @@ class BufferedPersistentDict(PersistentDict):
             An iterator over the items.
         """
         for keys, elements in self.storage.items():
-            for key, elt in zip(keys, elements):
-                yield key, elt
+            yield from zip(keys, elements)
 
     def delete(self) -> None:
         """Delete the storage directory and its contents."""

anemoi/datasets/create/size.py

@@ -10,8 +10,6 @@
 
 import logging
 import os
-from typing import Dict
-from typing import Optional
 
 import tqdm
 from anemoi.utils.humanize import bytes_to_human
@@ -19,7 +17,7 @@ from anemoi.utils.humanize import bytes_to_human
 LOG = logging.getLogger(__name__)
 
 
-def compute_directory_sizes(path: str) -> Optional[Dict[str, int]]:
+def compute_directory_sizes(path: str) -> dict[str, int] | None:
     """Computes the total size and number of files in a directory.
 
     Parameters

anemoi/datasets/create/sources/accumulations.py

@@ -10,14 +10,9 @@
 import datetime
 import logging
 import warnings
+from collections.abc import Generator
 from copy import deepcopy
 from typing import Any
-from typing import Dict
-from typing import Generator
-from typing import List
-from typing import Optional
-from typing import Tuple
-from typing import Union
 
 import earthkit.data as ekd
 import numpy as np
@@ -66,10 +61,10 @@ class Accumulation:
         date: int,
         time: int,
         number: int,
-        step: List[int],
+        step: list[int],
         frequency: int,
-        accumulations_reset_frequency: Optional[int] = None,
-        user_date: Optional[str] = None,
+        accumulations_reset_frequency: int | None = None,
+        user_date: str | None = None,
         **kwargs: Any,
     ) -> None:
         """Initialises an Accumulation instance.
@@ -103,10 +98,10 @@ class Accumulation:
         self.time = time
         self.steps = step
         self.number = number
-        self.values: Optional[NDArray[None]] = None
+        self.values: NDArray[None] | None = None
         self.seen = set()
-        self.startStep: Optional[int] = None
-        self.endStep: Optional[int] = None
+        self.startStep: int | None = None
+        self.endStep: int | None = None
         self.done = False
         self.frequency = frequency
         self.accumulations_reset_frequency = accumulations_reset_frequency
@@ -114,7 +109,7 @@ class Accumulation:
         self.user_date = user_date
 
     @property
-    def key(self) -> Tuple[str, int, int, List[int], int]:
+    def key(self) -> tuple[str, int, int, list[int], int]:
         """Returns the key for the accumulation."""
         return (self.param, self.date, self.time, self.steps, self.number)
 
@@ -235,15 +230,15 @@ class Accumulation:
     def mars_date_time_steps(
         cls,
         *,
-        dates: List[datetime.datetime],
+        dates: list[datetime.datetime],
         step1: int,
         step2: int,
-        frequency: Optional[int],
-        base_times: List[int],
+        frequency: int | None,
+        base_times: list[int],
         adjust_step: bool,
-        accumulations_reset_frequency: Optional[int],
-        user_date: Optional[str],
-    ) -> Generator[Tuple[int, int, Tuple[int, ...]], None, None]:
+        accumulations_reset_frequency: int | None,
+        user_date: str | None,
+    ) -> Generator[tuple[int, int, tuple[int, ...]], None, None]:
         """Generates MARS date-time steps.
 
         Parameters
@@ -327,11 +322,11 @@ class Accumulation:
         step1: int,
         step2: int,
         add_step: int,
-        frequency: Optional[int],
-        accumulations_reset_frequency: Optional[int],
-        user_date: Optional[str],
-        requested_date: Optional[datetime.datetime] = None,
-    ) -> Tuple[int, int, Tuple[int, ...]]:
+        frequency: int | None,
+        accumulations_reset_frequency: int | None,
+        user_date: str | None,
+        requested_date: datetime.datetime | None = None,
+    ) -> tuple[int, int, tuple[int, ...]]:
         """Generates a MARS date-time step.
 
         Parameters
@@ -364,7 +359,7 @@ class Accumulation:
 class AccumulationFromStart(Accumulation):
     """Class to handle data accumulation from the start of the forecast."""
 
-    def adjust_steps(self, startStep: int, endStep: int) -> Tuple[int, int]:
+    def adjust_steps(self, startStep: int, endStep: int) -> tuple[int, int]:
         """Adjusts the start and end steps.
 
         Parameters
@@ -427,11 +422,11 @@ class AccumulationFromStart(Accumulation):
         step1: int,
         step2: int,
         add_step: int,
-        frequency: Optional[int],
-        accumulations_reset_frequency: Optional[int],
-        user_date: Optional[str],
-        requested_date: Optional[datetime.datetime] = None,
-    ) -> Tuple[int, int, Tuple[int, ...]]:
+        frequency: int | None,
+        accumulations_reset_frequency: int | None,
+        user_date: str | None,
+        requested_date: datetime.datetime | None = None,
+    ) -> tuple[int, int, tuple[int, ...]]:
         """Generates a MARS date-time step.
 
         Parameters
@@ -518,10 +513,10 @@ class AccumulationFromLastStep(Accumulation):
         step2: int,
         add_step: int,
         frequency: int,
-        accumulations_reset_frequency: Optional[int],
-        user_date: Optional[str] = None,
-        requested_date: Optional[datetime.datetime] = None,
-    ) -> Tuple[int, int, Tuple[int, ...]]:
+        accumulations_reset_frequency: int | None,
+        user_date: str | None = None,
+        requested_date: datetime.datetime | None = None,
+    ) -> tuple[int, int, tuple[int, ...]]:
         """Generates a MARS date-time step.
 
         Parameters
@@ -568,7 +563,7 @@ class AccumulationFromLastStep(Accumulation):
 class AccumulationFromLastReset(Accumulation):
     """Class to handle data accumulation from the last step of the forecast."""
 
-    def adjust_steps(self, startStep: int, endStep: int) -> Tuple[int, int]:
+    def adjust_steps(self, startStep: int, endStep: int) -> tuple[int, int]:
         """Adjusts the start and end steps.
 
         Parameters
@@ -588,7 +583,7 @@ class AccumulationFromLastReset(Accumulation):
     @classmethod
     def _adjust_steps(
         self, startStep: int, endStep: int, frequency: int, accumulations_reset_frequency: int
-    ) -> Tuple[int, int]:
+    ) -> tuple[int, int]:
         """Adjusts the start and end steps.
 
         Parameters
@@ -620,7 +615,7 @@ class AccumulationFromLastReset(Accumulation):
         base_date: datetime.datetime,
         frequency: int,
         accumulations_reset_frequency: int,
-    ) -> Tuple[int, int]:
+    ) -> tuple[int, int]:
         """Calculates the steps for accumulation.
 
         Parameters
@@ -704,10 +699,10 @@ class AccumulationFromLastReset(Accumulation):
         step2: int,
         add_step: int,
         frequency: int,
-        accumulations_reset_frequency: Optional[int],
-        user_date: Optional[str],
-        requested_date: Optional[datetime.datetime] = None,
-    ) -> Tuple[int, int, Tuple[int, ...]]:
+        accumulations_reset_frequency: int | None,
+        user_date: str | None,
+        requested_date: datetime.datetime | None = None,
+    ) -> tuple[int, int, tuple[int, ...]]:
         """Generates a MARS date-time step.
 
         Parameters
@@ -776,15 +771,15 @@ def _identity(x: Any) -> Any:
 
 def _compute_accumulations(
     context: Any,
-    dates: List[datetime.datetime],
-    request: Dict[str, Any],
-    user_accumulation_period: Union[int, Tuple[int, int]] = 6,
-    data_accumulation_period: Optional[int] = None,
-    accumulations_reset_frequency: Optional[int] = None,
-    user_date: Optional[str] = None,
+    dates: list[datetime.datetime],
+    request: dict[str, Any],
+    user_accumulation_period: int | tuple[int, int] = 6,
+    data_accumulation_period: int | None = None,
+    accumulations_reset_frequency: int | None = None,
+    user_date: str | None = None,
     patch: Any = _identity,
-    base_times: Optional[List[int]] = None,
-    use_cdsapi_dataset: Optional[str] = None,
+    base_times: list[int] | None = None,
+    use_cdsapi_dataset: str | None = None,
 ) -> Any:
     """Computes accumulations based on the provided parameters.
 
@@ -933,7 +928,7 @@ def _compute_accumulations(
     return ds
 
 
-def _to_list(x: Union[List[Any], Tuple[Any], Any]) -> List[Any]:
+def _to_list(x: list[Any] | tuple[Any] | Any) -> list[Any]:
     """Converts the input to a list if it is not already a list or tuple.
 
     Parameters
@@ -951,7 +946,7 @@ def _to_list(x: Union[List[Any], Tuple[Any], Any]) -> List[Any]:
     return [x]
 
 
-def _scda(request: Dict[str, Any]) -> Dict[str, Any]:
+def _scda(request: dict[str, Any]) -> dict[str, Any]:
     """Modifies the request stream based on the time.
 
     Parameters
@@ -973,7 +968,7 @@ def _scda(request: Dict[str, Any]) -> Dict[str, Any]:
 
 @legacy_source(__file__)
 def accumulations(
-    context: Any, dates: List[datetime.datetime], use_cdsapi_dataset: Optional[str] = None, **request: Any
+    context: Any, dates: list[datetime.datetime], use_cdsapi_dataset: str | None = None, **request: Any
 ) -> Any:
     """Computes accumulations based on the provided context, dates, and request parameters.
 

anemoi/datasets/create/sources/accumulations2.py

@@ -12,10 +12,6 @@ import logging
 from abc import abstractmethod
 from copy import deepcopy
 from typing import Any
-from typing import Dict
-from typing import List
-from typing import Tuple
-from typing import Union
 
 import earthkit.data as ekd
 import numpy as np
@@ -477,8 +473,8 @@ class Accumulator:
 
 def _compute_accumulations(
     context: Any,
-    dates: List[datetime.datetime],
-    request: Dict[str, Any],
+    dates: list[datetime.datetime],
+    request: dict[str, Any],
     user_accumulation_period: datetime.timedelta,
     # data_accumulation_period: Optional[int] = None,
     # patch: Any = _identity,
@@ -565,7 +561,7 @@ def _compute_accumulations(
     return ds
 
 
-def _to_list(x: Union[List[Any], Tuple[Any], Any]) -> List[Any]:
+def _to_list(x: list[Any] | tuple[Any] | Any) -> list[Any]:
     """Converts the input to a list if it is not already a list or tuple.
 
     Parameters
@@ -583,7 +579,7 @@ def _to_list(x: Union[List[Any], Tuple[Any], Any]) -> List[Any]:
     return [x]
 
 
-def _scda(request: Dict[str, Any]) -> Dict[str, Any]:
+def _scda(request: dict[str, Any]) -> dict[str, Any]:
     """Modifies the request stream based on the time.
 
     Parameters

anemoi/datasets/create/sources/constants.py

@@ -8,8 +8,6 @@
 # nor does it submit to any jurisdiction.
 
 from typing import Any
-from typing import Dict
-from typing import List
 
 from earthkit.data import from_source
 
@@ -17,7 +15,7 @@ from .legacy import legacy_source
 
 
 @legacy_source(__file__)
-def constants(context: Any, dates: List[str], template: Dict[str, Any], param: str) -> Any:
+def constants(context: Any, dates: list[str], template: dict[str, Any], param: str) -> Any:
     """Deprecated function to retrieve constants data.
 
     Parameters