diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index ad8c19e..78bad1a 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -56,7 +56,7 @@ repos: args: [--wrap, "88"] files: (README\.md) - repo: https://github.com/crate-ci/typos - rev: v1.46.0 + rev: v1 hooks: - id: typos - repo: meta diff --git a/pyproject.toml b/pyproject.toml index d2351a0..f788463 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -50,7 +50,7 @@ test = [ ] typing = [ "pytask-parallel", - "ty>=0.0.8", + "ty>=0.0.8,<0.0.34", {include-group = "coiled"}, {include-group = "dask"}, ] diff --git a/src/pytask_parallel/execute.py b/src/pytask_parallel/execute.py index 3001d69..52cc51d 100644 --- a/src/pytask_parallel/execute.py +++ b/src/pytask_parallel/execute.py @@ -44,7 +44,7 @@ def _get_task_from_dag(session: Session, task_name: str) -> PTask: """Get a task from the pre- and post-pytask 0.6 DAG representations for compat.""" node = session.dag.nodes[task_name] - task = node["task"] if isinstance(node, dict) else node + task = cast("Any", node)["task"] if isinstance(node, dict) else node if not isinstance(task, PTask): msg = f"Expected {task_name!r} to resolve to a task."