Fix.

Author: tobiasraabe

tests/test_collect_command.py

@@ -619,14 +619,15 @@ def test_more_nested_pytree_and_python_node_as_return(runner, tmp_path):
     from typing import Any
     from typing_extensions import Annotated
     from pytask import PythonNode
+    from typing import Dict
 
     nodes = [
         PythonNode(name="dict"),
         (PythonNode(name="tuple1"), PythonNode(name="tuple2")),
         PythonNode(name="int")
     ]
 
-    def task_example() -> Annotated[dict[str, str], nodes]:
+    def task_example() -> Annotated[Dict[str, str], nodes]:
         return [{"first": "a", "second": "b"}, (1, 2), 1]
     """
     tmp_path.joinpath("task_module.py").write_text(textwrap.dedent(source))

tests/test_execute.py

@@ -675,8 +675,9 @@ def test_pytree_and_python_node_as_return(runner, tmp_path):
     from typing import Any
     from typing_extensions import Annotated
     from pytask import PythonNode
+    from typing import Dict
 
-    def task_example() -> Annotated[dict[str, str], PythonNode(name="result")]:
+    def task_example() -> Annotated[Dict[str, str], PythonNode(name="result")]:
         return {"first": "a", "second": "b"}
     """
     tmp_path.joinpath("task_module.py").write_text(textwrap.dedent(source))
@@ -691,14 +692,15 @@ def test_more_nested_pytree_and_python_node_as_return(runner, tmp_path):
     from typing import Any
     from typing_extensions import Annotated
     from pytask import PythonNode
+    from typing import Dict
 
     nodes = [
         PythonNode(name="dict"),
         (PythonNode(name="tuple1"), PythonNode(name="tuple2")),
         PythonNode(name="int")
     ]
 
-    def task_example() -> Annotated[dict[str, str], nodes]:
+    def task_example() -> Annotated[Dict[str, str], nodes]:
         return [{"first": "a", "second": "b"}, (1, 2), 1]
     """
     tmp_path.joinpath("task_module.py").write_text(textwrap.dedent(source))