improve annotated.

Author: tobiasraabe

src/_pytask/collect_utils.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 import itertools
+import sys
 import uuid
 import warnings
 from pathlib import Path
@@ -29,9 +30,12 @@
 from _pytask.typing import ProductType
 from attrs import define
 from attrs import field
-from typing_extensions import Annotated
 from typing_extensions import get_origin
 
+if sys.version_info >= (3, 9):
+    from typing import Annotated
+else:
+    from typing_extensions import Annotated
 
 if TYPE_CHECKING:
     from _pytask.session import Session

src/_pytask/task_utils.py

@@ -74,7 +74,7 @@ def task(
 
     .. code-block:: python
 
-        from typing_extensions import Annotated
+        from typing import Annotated
         from pytask import task
 
         @task