Fix linting issues

Author: tobiasraabe

docs_src/how_to_guides/migrating_from_scripts_to_pytask_5.py

@@ -1,14 +1,13 @@
 from pathlib import Path
 from typing import Annotated
-from typing import Optional
 
 import pandas as pd
 
 from pytask import Product
 
 
 def task_merge_data(
-    paths_to_input_data: Optional[dict[str, Path]] = None,
+    paths_to_input_data: dict[str, Path] | None = None,
     path_to_merged_data: Annotated[Path, Product] = Path("merged_data.pkl"),
 ) -> None:
     if paths_to_input_data is None:

docs_src/how_to_guides/writing_custom_nodes_example_3_py38.py

@@ -2,7 +2,6 @@
 import pickle
 from pathlib import Path
 from typing import Any
-from typing import Optional
 
 from pytask import hash_value
 
@@ -24,8 +23,8 @@ class PickleNode:
     def __init__(
         self,
         name: str = "",
-        path: Optional[Path] = None,
-        attributes: Optional[dict[Any, Any]] = None,
+        path: Path | None = None,
+        attributes: dict[Any, Any] | None = None,
     ) -> None:
         self.name = name
         self.path = path
@@ -45,7 +44,7 @@ def from_path(cls, path: Path) -> "PickleNode":
             raise ValueError(msg)
         return cls(name=path.as_posix(), path=path)
 
-    def state(self) -> Optional[str]:
+    def state(self) -> str | None:
         """Return the modification timestamp as the state."""
         if self.path.exists():
             return str(self.path.stat().st_mtime)

src/_pytask/_inspect.py

@@ -1,143 +1,6 @@
 from __future__ import annotations
 
-import functools
-import sys
-import types
-from typing import TYPE_CHECKING
-from typing import Any
-from typing import Callable
-
-if TYPE_CHECKING:
-    from collections.abc import Mapping
-
 __all__ = ["get_annotations"]
 
 
-if sys.version_info >= (3, 10):  # pragma: no cover
-    from inspect import get_annotations
-else:  # pragma: no cover
-
-    def get_annotations(  # noqa: C901, PLR0912, PLR0915
-        obj: Callable[..., object] | type[Any] | types.ModuleType,
-        *,
-        globals: Mapping[str, Any] | None = None,  # noqa: A002
-        locals: Mapping[str, Any] | None = None,  # noqa: A002
-        eval_str: bool = False,
-    ) -> dict[str, Any]:
-        """Compute the annotations dict for an object.
-
-        obj may be a callable, class, or module.
-        Passing in an object of any other type raises TypeError.
-
-        Returns a dict.  get_annotations() returns a new dict every time
-        it's called; calling it twice on the same object will return two
-        different but equivalent dicts.
-
-        This function handles several details for you:
-
-        * If eval_str is true, values of type str will
-            be un-stringized using eval().  This is intended
-            for use with stringized annotations
-            ("from __future__ import annotations").
-        * If obj doesn't have an annotations dict, returns an
-            empty dict.  (Functions and methods always have an
-            annotations dict; classes, modules, and other types of
-            callables may not.)
-        * Ignores inherited annotations on classes.  If a class
-            doesn't have its own annotations dict, returns an empty dict.
-        * All accesses to object members and dict values are done
-            using getattr() and dict.get() for safety.
-        * Always, always, always returns a freshly-created dict.
-
-        eval_str controls whether or not values of type str are replaced
-        with the result of calling eval() on those values:
-
-        * If eval_str is true, eval() is called on values of type str.
-        * If eval_str is false (the default), values of type str are unchanged.
-
-        globals and locals are passed in to eval(); see the documentation
-        for eval() for more information.  If either globals or locals is
-        None, this function may replace that value with a context-specific
-        default, contingent on type(obj):
-
-        * If obj is a module, globals defaults to obj.__dict__.
-        * If obj is a class, globals defaults to
-            sys.modules[obj.__module__].__dict__ and locals
-            defaults to the obj class namespace.
-        * If obj is a callable, globals defaults to obj.__globals__,
-            although if obj is a wrapped function (using
-            functools.update_wrapper()) it is first unwrapped.
-        """
-        if isinstance(obj, type):
-            # class
-            obj_dict = getattr(obj, "__dict__", None)
-            if obj_dict and hasattr(obj_dict, "get"):
-                ann = obj_dict.get("__annotations__", None)
-                if isinstance(ann, types.GetSetDescriptorType):
-                    ann = None
-            else:
-                ann = None
-
-            obj_globals = None
-            module_name = getattr(obj, "__module__", None)
-            if module_name:
-                module = sys.modules.get(module_name, None)
-                if module:
-                    obj_globals = getattr(module, "__dict__", None)
-            obj_locals = dict(vars(obj))
-            unwrap = obj
-        elif isinstance(obj, types.ModuleType):
-            # module
-            ann = getattr(obj, "__annotations__", None)
-            obj_globals = obj.__dict__
-            obj_locals = None
-            unwrap = None
-        elif callable(obj):
-            # this includes types.Function, types.BuiltinFunctionType,
-            # types.BuiltinMethodType, functools.partial, functools.singledispatch,
-            # "class funclike" from Lib/test/test_inspect... on and on it goes.
-            ann = getattr(obj, "__annotations__", None)
-            obj_globals = getattr(obj, "__globals__", None)
-            obj_locals = None
-            unwrap = obj
-        else:
-            msg = f"{obj!r} is not a module, class, or callable."
-            raise TypeError(msg)
-
-        if ann is None:
-            return {}
-
-        if not isinstance(ann, dict):
-            msg = f"{obj!r}.__annotations__ is neither a dict nor None"
-            raise ValueError(msg)  # noqa: TRY004
-
-        if not ann:
-            return {}
-
-        if not eval_str:
-            return dict(ann)
-
-        if unwrap is not None:
-            while True:
-                if hasattr(unwrap, "__wrapped__"):
-                    unwrap = unwrap.__wrapped__
-                    continue
-                if isinstance(unwrap, functools.partial):
-                    unwrap = unwrap.func
-                    continue
-                break
-            if hasattr(unwrap, "__globals__"):
-                obj_globals = unwrap.__globals__
-
-        if globals is None:
-            globals = obj_globals  # noqa: A001
-        if locals is None:
-            locals = obj_locals  # noqa: A001
-
-        eval_func = eval
-        return {
-            key: value
-            if not isinstance(value, str)
-            else eval_func(value, globals, locals)
-            for key, value in ann.items()
-        }
+from inspect import get_annotations

src/_pytask/build.py

@@ -8,7 +8,6 @@
 from pathlib import Path
 from typing import TYPE_CHECKING
 from typing import Any
-from typing import Callable
 from typing import Literal
 
 import click
@@ -35,6 +34,7 @@
 from _pytask.traceback import Traceback
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
     from collections.abc import Iterable
     from typing import NoReturn
 

src/_pytask/cache.py

@@ -6,14 +6,17 @@
 import hashlib
 import inspect
 from inspect import FullArgSpec
+from typing import TYPE_CHECKING
 from typing import Any
-from typing import Callable
 
 from attrs import define
 from attrs import field
 
 from _pytask._hashlib import hash_value
 
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
 
 @define
 class CacheInfo:

src/_pytask/coiled_utils.py

@@ -1,10 +1,13 @@
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
 from typing import Any
-from typing import Callable
 
 from attrs import define
 
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
 try:
     from coiled.function import Function
 except ImportError:

src/_pytask/collect_utils.py

@@ -6,7 +6,6 @@
 from typing import TYPE_CHECKING
 from typing import Annotated
 from typing import Any
-from typing import Callable
 from typing import get_origin
 
 import attrs
@@ -25,6 +24,7 @@
 from _pytask.typing import no_default
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
     from pathlib import Path
 
     from _pytask.session import Session

src/_pytask/console.py

@@ -8,7 +8,6 @@
 from pathlib import Path
 from typing import TYPE_CHECKING
 from typing import Any
-from typing import Callable
 
 from rich.console import Console
 from rich.console import RenderableType
@@ -29,6 +28,7 @@
 from _pytask.path import shorten_path
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
     from collections.abc import Iterable
     from collections.abc import Sequence
     from enum import Enum

src/_pytask/dag_command.py

@@ -216,7 +216,7 @@ def _shorten_node_labels(dag: nx.DiGraph, paths: list[Path]) -> nx.DiGraph:
     node_names = dag.nodes
     short_names = reduce_names_of_multiple_nodes(node_names, dag, paths)
     short_names = [i.plain if isinstance(i, Text) else i for i in short_names]  # type: ignore[attr-defined]
-    old_to_new = dict(zip(node_names, short_names))
+    old_to_new = dict(zip(node_names, short_names, strict=False))
     return nx.relabel_nodes(dag, old_to_new)
 
 

src/_pytask/execute.py

@@ -287,7 +287,7 @@ def pytask_execute_task(session: Session, task: PTask) -> bool:
 
         nodes = tree_leaves(task.produces["return"])
         values = structure_return.flatten_up_to(out)
-        for node, value in zip(nodes, values):
+        for node, value in zip(nodes, values, strict=False):
             if not isinstance(node, PProvisionalNode):
                 node.save(value)