Import many stuff from typing instead of typing_extensions.

Author: tobiasraabe

README.md

@@ -82,7 +82,7 @@ with `task_`. Here is an example.
 from pathlib import Path
 
 from pytask import Product
-from typing_extensions import Annotated
+from typing import Annotated
 
 
 def task_hello_earth(path: Annotated[Path, Product] = Path("hello_earth.txt")):

docs/source/how_to_guides/the_data_catalog.md

@@ -110,7 +110,7 @@ The task could look like this.
 ```python
 from pathlib import Path
 from pytask import task
-from typing_extensions import Annotated
+from typing import Annotated
 
 from my_project.config import DATA_NAMES
 from my_project.config import MODEL_NAMES

docs/source/how_to_guides/writing_custom_nodes.md

@@ -142,9 +142,9 @@ databases. [^kedro]
 
 ## References
 
-[^structural-subtyping]: Structural subtyping is similar to ABCs an approach in Python to enforce interfaces, but
-    it can be considered more pythonic since it is closer to duck typing. Hynek Schlawack
-    wrote a comprehensive
+[^structural-subtyping]: Structural subtyping is similar to ABCs an approach in Python to enforce interfaces,
+    but it can be considered more pythonic since it is closer to duck typing. Hynek
+    Schlawack wrote a comprehensive
     [guide on subclassing](https://hynek.me/articles/python-subclassing-redux/) that
     features protocols under "Type 2". Glyph wrote an introduction to protocols called
     [I want a new duck](https://glyph.twistedmatrix.com/2020/07/new-duck.html).

docs/source/reference_guides/api.md

@@ -329,7 +329,7 @@ resolution and execution.
 
     >>> from pathlib import Path
     >>> from pytask import Product
-    >>> from typing_extensions import Annotated
+    >>> from typing import Annotated
     >>> def task_example(path: Annotated[Path, Product]) -> None:
     ...     path.write_text("Hello, World!")
 

docs_src/how_to_guides/bp_structure_of_task_files.py

@@ -1,8 +1,8 @@
 from pathlib import Path
+from typing import Annotated
 
 import pandas as pd
 from checks import perform_general_checks_on_data
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/how_to_guides/capturing_warnings_1.py

@@ -1,7 +1,7 @@
 from pathlib import Path
+from typing import Annotated
 
 import pandas as pd
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/how_to_guides/capturing_warnings_2.py

@@ -1,7 +1,7 @@
 from pathlib import Path
+from typing import Annotated
 
 import pandas as pd
-from typing_extensions import Annotated
 
 import pytask
 from pytask import Product

docs_src/how_to_guides/migrating_from_scripts_to_pytask_4.py

@@ -1,8 +1,8 @@
 # Content of task_data_management.py
 from pathlib import Path
+from typing import Annotated
 
 import pandas as pd
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/how_to_guides/migrating_from_scripts_to_pytask_5.py

@@ -1,8 +1,8 @@
 from pathlib import Path
+from typing import Annotated
 from typing import Optional
 
 import pandas as pd
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/how_to_guides/provisional_products.py

@@ -1,7 +1,7 @@
 from pathlib import Path
+from typing import Annotated
 
 import httpx
-from typing_extensions import Annotated
 
 from pytask import DirectoryNode
 from pytask import Product

docs_src/how_to_guides/provisional_task.py

@@ -1,6 +1,5 @@
 from pathlib import Path
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 from pytask import DirectoryNode
 

docs_src/how_to_guides/provisional_task_generator.py

@@ -1,6 +1,5 @@
 from pathlib import Path
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 from pytask import DirectoryNode
 from pytask import task

docs_src/how_to_guides/using_task_returns_example_1_py38.py

@@ -1,6 +1,5 @@
 from pathlib import Path
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 
 def task_create_file() -> Annotated[str, Path("file.txt")]:

docs_src/how_to_guides/using_task_returns_example_3_py38.py

@@ -1,6 +1,5 @@
 from pathlib import Path
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 
 def task_create_files() -> Annotated[str, (Path("file1.txt"), Path("file2.txt"))]:

docs_src/how_to_guides/using_task_returns_example_4_py38.py

@@ -1,7 +1,6 @@
+from typing import Annotated
 from typing import Any
 
-from typing_extensions import Annotated
-
 from pytask import PythonNode
 
 nodes = [

docs_src/how_to_guides/writing_custom_nodes_example_2_py38.py

@@ -1,7 +1,7 @@
 from pathlib import Path
+from typing import Annotated
 
 import pandas as pd
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/how_to_guides/writing_custom_nodes_example_2_py38_return.py

@@ -1,7 +1,7 @@
 from pathlib import Path
+from typing import Annotated
 
 import pandas as pd
-from typing_extensions import Annotated
 
 
 class PickleNode: ...

docs_src/tutorials/defining_dependencies_products_dependencies_py38.py

@@ -1,9 +1,9 @@
 from pathlib import Path
+from typing import Annotated
 
 import matplotlib.pyplot as plt
 import pandas as pd
 from my_project.config import BLD
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/tutorials/defining_dependencies_products_multiple1_produces.py

@@ -1,5 +1,4 @@
 from pathlib import Path
-from typing import Dict
 
 from my_project.config import BLD
 
@@ -9,5 +8,5 @@
 def task_plot_data(
     path_to_data_0: Path = BLD / "data_0.pkl",
     path_to_data_1: Path = BLD / "data_1.pkl",
-    produces: Dict[str, Path] = _PRODUCTS,
+    produces: dict[str, Path] = _PRODUCTS,
 ) -> None: ...

docs_src/tutorials/defining_dependencies_products_multiple1_py38.py

@@ -1,7 +1,7 @@
 from pathlib import Path
+from typing import Annotated
 
 from my_project.config import BLD
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/tutorials/defining_dependencies_products_multiple2_py38.py

@@ -1,8 +1,7 @@
 from pathlib import Path
-from typing import Dict
+from typing import Annotated
 
 from my_project.config import BLD
-from typing_extensions import Annotated
 
 from pytask import Product
 
@@ -11,6 +10,6 @@
 
 
 def task_plot_data(
-    path_to_data: Dict[str, Path] = _DEPENDENCIES,
-    path_to_plots: Annotated[Dict[str, Path], Product] = _PRODUCTS,
+    path_to_data: dict[str, Path] = _DEPENDENCIES,
+    path_to_plots: Annotated[dict[str, Path], Product] = _PRODUCTS,
 ) -> None: ...

docs_src/tutorials/defining_dependencies_products_products_py38.py

@@ -1,9 +1,9 @@
 from pathlib import Path
+from typing import Annotated
 
 import numpy as np
 import pandas as pd
 from my_project.config import BLD
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/tutorials/defining_dependencies_products_relative_py38.py

@@ -1,6 +1,5 @@
 from pathlib import Path
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 from pytask import Product
 

docs_src/tutorials/repeating_tasks_with_different_inputs1_py38.py

@@ -1,6 +1,5 @@
 from pathlib import Path
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 from pytask import Product
 from pytask import task

docs_src/tutorials/repeating_tasks_with_different_inputs2_py38.py

@@ -1,7 +1,7 @@
 from pathlib import Path
+from typing import Annotated
 
 from my_project.config import SRC
-from typing_extensions import Annotated
 
 from pytask import Product
 from pytask import task

docs_src/tutorials/repeating_tasks_with_different_inputs3_produces.py

@@ -1,11 +1,10 @@
 from pathlib import Path
-from typing import Tuple
 
 from pytask import task
 
 for seed in ((0,), (1,)):
 
     @task
     def task_create_random_data(
-        produces: Path = Path(f"data_{seed[0]}.pkl"), seed: Tuple[int] = seed
+        produces: Path = Path(f"data_{seed[0]}.pkl"), seed: tuple[int] = seed
     ) -> None: ...

docs_src/tutorials/repeating_tasks_with_different_inputs3_py38.py

@@ -1,7 +1,5 @@
 from pathlib import Path
-from typing import Tuple
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 from pytask import Product
 from pytask import task
@@ -10,6 +8,6 @@
 
     @task
     def task_create_random_data(
-        seed: Tuple[int] = seed,
+        seed: tuple[int] = seed,
         path_to_data: Annotated[Path, Product] = Path(f"data_{seed[0]}.pkl"),
     ) -> None: ...

docs_src/tutorials/repeating_tasks_with_different_inputs4_py38.py

@@ -1,6 +1,5 @@
 from pathlib import Path
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 from pytask import Product
 from pytask import task

docs_src/tutorials/repeating_tasks_with_different_inputs5_py38.py

@@ -1,8 +1,7 @@
 from pathlib import Path
+from typing import Annotated
 from typing import NamedTuple
 
-from typing_extensions import Annotated
-
 from pytask import Product
 from pytask import task
 

docs_src/tutorials/skipping_tasks_example_1.py

@@ -1,6 +1,5 @@
 from pathlib import Path
-
-from typing_extensions import Annotated
+from typing import Annotated
 
 import pytask
 from pytask import Product

docs_src/tutorials/using_a_data_catalog_2_py38.py

@@ -1,7 +1,8 @@
+from typing import Annotated
+
 import numpy as np
 import pandas as pd
 from my_project.config import data_catalog
-from typing_extensions import Annotated
 
 from pytask import PickleNode
 from pytask import Product

docs_src/tutorials/using_a_data_catalog_3_py38.py

@@ -1,10 +1,10 @@
 from pathlib import Path
+from typing import Annotated
 
 import matplotlib.pyplot as plt
 import pandas as pd
 from my_project.config import BLD
 from my_project.config import data_catalog
-from typing_extensions import Annotated
 
 from pytask import Product
 

docs_src/tutorials/using_a_data_catalog_5_py38.py

@@ -1,8 +1,8 @@
 from pathlib import Path
+from typing import Annotated
 
 import pandas as pd
 from my_project.config import data_catalog
-from typing_extensions import Annotated
 
 from pytask import PickleNode
 from pytask import Product

docs_src/tutorials/write_a_task_py38.py

@@ -1,10 +1,10 @@
 # Content of task_data_preparation.py.
 from pathlib import Path
+from typing import Annotated
 
 import numpy as np
 import pandas as pd
 from my_project.config import BLD
-from typing_extensions import Annotated
 
 from pytask import Product
 

scripts/update_plugin_list.py

@@ -34,14 +34,17 @@
 import pathlib
 import re
 from textwrap import indent
-from typing import Generator
+from typing import TYPE_CHECKING
 
 import httpx
 import packaging.version
 import tabulate
 import wcwidth
 from tqdm import tqdm
 
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
 _FILE_HEAD = r"""
 .. _plugin-list:
 

src/_pytask/_inspect.py

@@ -3,9 +3,12 @@
 import functools
 import sys
 import types
+from typing import TYPE_CHECKING
 from typing import Any
 from typing import Callable
-from typing import Mapping
+
+if TYPE_CHECKING:
+    from collections.abc import Mapping
 
 __all__ = ["get_annotations"]
 

src/_pytask/build.py

@@ -9,7 +9,6 @@
 from typing import TYPE_CHECKING
 from typing import Any
 from typing import Callable
-from typing import Iterable
 from typing import Literal
 
 import click
@@ -36,6 +35,7 @@
 from _pytask.traceback import Traceback
 
 if TYPE_CHECKING:
+    from collections.abc import Iterable
     from typing import NoReturn
 
     from _pytask.node_protocols import PTask