Cleanup

src/metldata/builtin_transformations/infer_relations/path/path_str.py

@@ -17,7 +17,6 @@
 """Data models"""
 
 import re
-from typing import Optional
 
 from metldata.builtin_transformations.infer_relations.path.path_elements import (
     RelationPathElement,
@@ -107,7 +106,7 @@ def get_target_class(*, path_str: str) -> str:
     return match.group(1)
 
 
-def split_first_element(*, path_str: str) -> tuple[str, Optional[str]]:
+def split_first_element(*, path_str: str) -> tuple[str, str | None]:
     """Return a tuple of the first element and the remaining path string.
     Thereby, the target class of the first element is set as the source class of the
     remaining path.

src/metldata/custom_types.py

@@ -16,9 +16,7 @@
 
 """A collection of custom types."""
 
-from typing import Any
-
-from typing_extensions import TypeAlias
+from typing import Any, TypeAlias
 
 Json: TypeAlias = dict[str, Any]
 

src/metldata/transform/base.py

@@ -18,9 +18,9 @@
 
 from abc import ABC, abstractmethod
 from collections import defaultdict
-from collections.abc import Generator
+from collections.abc import Callable, Generator
 from graphlib import CycleError, TopologicalSorter
-from typing import Callable, Generic, Optional, TypeVar
+from typing import Generic, TypeAlias, TypeVar
 
 from pydantic import (
     BaseModel,
@@ -34,7 +34,6 @@
 from schemapack.spec.custom_types import ClassName, ResourceId
 from schemapack.spec.datapack import DataPack
 from schemapack.spec.schemapack import SchemaPack
-from typing_extensions import TypeAlias
 
 from metldata.custom_types import Json
 
@@ -148,7 +147,7 @@ class WorkflowStepBase(BaseModel, ABC):
 
     model_config = ConfigDict(frozen=True)
     description: str = Field(..., description="A description of the step.")
-    input: Optional[str] = Field(
+    input: str | None = Field(
         ...,
         description=(
             "The name of the workflow step from which the output is used as input"

tests/builtin_transformations/infer_relations/path/test_path_str.py

@@ -17,7 +17,6 @@
 """Test the path_str module."""
 
 from contextlib import nullcontext
-from typing import Optional
 
 import pytest
 
@@ -132,7 +131,7 @@ def test_get_target_class(path_str: str, expected_target_class: str):
     ],
 )
 def test_split_first_element(
-    path_str: str, expected_first_element: str, expected_remaining_path: Optional[str]
+    path_str: str, expected_first_element: str, expected_remaining_path: str | None
 ):
     """Test the split_first_element method."""
     observed_first_element, observed_remaining_path = split_first_element(