taskcluster-taskgraph 16.2.0__tar.gz → 17.0.0__tar.gz

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/CHANGELOG.md

@@ -1,4 +1,40 @@
 # Change Log
+
+## [17.0.0] - 2025-10-17
+
+### Changed
+
+- BREAKING CHANGE: Dropped support for Python 3.8
+- BREAKING CHANGE: `base_ref` parameter no longer gets reset to default branch by Decision task
+- BREAKING CHANGE: `base_rev` parameter no longer gets reset to merge-base by Decision task
+- BREAKING CHANGE: `run-task` script no longer fetches or checks out `base_ref` parameter
+- BREAKING CHANGE: `run-task` script no longer fetches all heads if `head_ref` was not specified
+
+### Added
+
+- `run-task` now supports shallow Git clones
+- Improved support for shallow clones in `repo.get_changed_files()`
+
+### Perf
+
+- Improvements to graph traversal logic and caching
+
+### Fixed
+
+- Bug in `retrigger-multiple` action's schema definition
+- `run-task` no longer fetches head ref twice in some scenarios
+- `run-task` avoids fetching tags in some scenarios when not necessary
+- `run-task` properly normalizes `head_ref` before checking it out
+- Appropriate exception now being caught in various `taskcluster` call sites
+
+## [16.2.1] - 2025-10-14
+
+### Fixed
+
+- `run-task` script now cleans up zombie processes
+- Fixed regression to `taskgraph.util.taskcluster.list_artifacts`
+- Fixed regression to `taskgraph.util.taskcluster.get_artifact`
+
 ## [16.2.0] - 2025-09-26
 
 ### Added

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: taskcluster-taskgraph
-Version: 16.2.0
+Version: 17.0.0
 Summary: Build taskcluster taskgraphs
 Project-URL: Repository, https://github.com/taskcluster/taskgraph
 Project-URL: Issues, https://github.com/taskcluster/taskgraph/issues
@@ -9,14 +9,13 @@ License-File: LICENSE
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Environment :: Console
 Classifier: License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Requires-Dist: appdirs>=1.4
 Requires-Dist: cookiecutter~=2.1
 Requires-Dist: json-e>=2.7
@@ -26,7 +25,7 @@ Requires-Dist: redo>=2.0
 Requires-Dist: requests>=2.25
 Requires-Dist: slugid>=2.0
 Requires-Dist: taskcluster-urls>=11.0
-Requires-Dist: taskcluster>=55.0
+Requires-Dist: taskcluster>=91.0
 Requires-Dist: voluptuous>=0.12.1
 Provides-Extra: load-image
 Requires-Dist: zstandard; extra == 'load-image'

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/docs/reference/migrations.rst

@@ -3,6 +3,38 @@ Migration Guide
 
 This page can help when migrating Taskgraph across major versions.
 
+16.x -> 17.x
+------------
+
+* Upgrade to Python 3.9 or above.
+* The ``base_ref`` parameter could now be ``None`` if it wasn't explicitly
+  passed into the Decision task. If this was relied upon, do the following
+  instead:
+
+  .. code-block:: python
+
+     from taskgraph.util.vcs import get_repository
+
+     repo = get_repository()
+     base_ref = repo.default_branch
+
+* The ``base_rev`` parameter is no longer being reset to the merge-base. This
+  means in some cases, ``base_rev`` might not actually be an ancestor of
+  ``head_rev``. If you need the merge-base, you can use:
+
+  .. code-block:: python
+
+     from taskgraph.util.vcs import get_repository
+
+     repo = get_repository()
+     base_rev = repo.find_latest_common_revision(parameters["base_rev"], parameters["head_rev"])
+
+* The ``base_ref`` is no longer being fetched by ``run-task``. If you were
+  relying on it, you can run ``git fetch origin <base_ref>`` in a subprocess.
+* The ``run-task`` scripts no longer fetches all heads when ``head_ref`` isn't
+  specified. If this behaviour was relied upon, you can run
+  ``git fetch +refs/heads/*:refs/remotes/work/*`` in a subprocess.
+
 15.x -> 16.x
 ------------
 

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/pyproject.toml

@@ -1,7 +1,7 @@
 ### Project
 [project]
 name = "taskcluster-taskgraph"
-version = "16.2.0"
+version = "17.0.0"
 description = "Build taskcluster taskgraphs"
 readme = "README.rst"
 authors = [
@@ -11,7 +11,6 @@ classifiers = [
   "Development Status :: 5 - Production/Stable",
   "Environment :: Console",
   "License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)",
-  "Programming Language :: Python :: 3.8",
   "Programming Language :: Python :: 3.9",
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
@@ -19,7 +18,7 @@ classifiers = [
   "Programming Language :: Python :: 3.13",
   "Topic :: Software Development",
 ]
-requires-python = ">=3.8"
+requires-python = ">=3.9"
 dependencies = [
   "appdirs>=1.4",
   "cookiecutter~=2.1",
@@ -29,7 +28,7 @@ dependencies = [
   "redo>=2.0",
   "requests>=2.25",
   "slugid>=2.0",
-  "taskcluster>=55.0",
+  "taskcluster>=91.0",
   "taskcluster-urls>=11.0",
   "voluptuous>=0.12.1",
 ]

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/actions/cancel.py

@@ -7,6 +7,7 @@ import logging
 
 import requests
 
+from taskcluster import TaskclusterRestFailure
 from taskgraph.util.taskcluster import cancel_task
 
 from .registry import register_callback_action
@@ -27,8 +28,14 @@ def cancel_action(parameters, graph_config, input, task_group_id, task_id):
     # only cancel tasks with the level-specific schedulerId.
     try:
         cancel_task(task_id)
-    except requests.HTTPError as e:
-        if e.response.status_code == 409:
+    except (requests.HTTPError, TaskclusterRestFailure) as e:
+        status_code = None
+        if isinstance(e, requests.HTTPError):
+            status_code = e.response.status_code if e.response else None
+        elif isinstance(e, TaskclusterRestFailure):
+            status_code = e.status_code
+
+        if status_code == 409:
             # A 409 response indicates that this task is past its deadline.  It
             # cannot be cancelled at this time, but it's also not running
             # anymore, so we can ignore this error.

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/actions/cancel_all.py

@@ -9,6 +9,7 @@ from concurrent import futures
 
 import requests
 
+from taskcluster import TaskclusterRestFailure
 from taskgraph.util.taskcluster import (
     CONCURRENCY,
     cancel_task,
@@ -36,8 +37,14 @@ def cancel_all_action(parameters, graph_config, input, task_group_id, task_id):
         logger.info(f"Cancelling task {task_id}")
         try:
             cancel_task(task_id)
-        except requests.HTTPError as e:
-            if e.response.status_code == 409:
+        except (requests.HTTPError, TaskclusterRestFailure) as e:
+            status_code = None
+            if isinstance(e, requests.HTTPError):
+                status_code = e.response.status_code if e.response else None
+            elif isinstance(e, TaskclusterRestFailure):
+                status_code = e.status_code
+
+            if status_code == 409:
                 # A 409 response indicates that this task is past its deadline.  It
                 # cannot be cancelled at this time, but it's also not running
                 # anymore, so we can ignore this error.

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/actions/registry.py

@@ -30,13 +30,13 @@ def is_json(data):
     return True
 
 
-@functools.lru_cache(maxsize=None)
+@functools.cache
 def read_taskcluster_yml(filename):
     """Load and parse .taskcluster.yml, cached to save some time"""
     return yaml.load_yaml(filename)
 
 
-@functools.lru_cache(maxsize=None)
+@functools.cache
 def hash_taskcluster_yml(filename):
     """
     Generate a hash of the given .taskcluster.yml.  This is the first 10 digits

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/actions/retrigger.py

@@ -233,17 +233,20 @@ def _rerun_task(task_id, label):
             "requests": {
                 "type": "array",
                 "items": {
-                    "tasks": {
-                        "type": "array",
-                        "description": "An array of task labels",
-                        "items": {"type": "string"},
-                    },
-                    "times": {
-                        "type": "integer",
-                        "minimum": 1,
-                        "maximum": 100,
-                        "title": "Times",
-                        "description": "How many times to run each task.",
+                    "type": "object",
+                    "properties": {
+                        "tasks": {
+                            "type": "array",
+                            "description": "An array of task labels",
+                            "items": {"type": "string"},
+                        },
+                        "times": {
+                            "type": "integer",
+                            "minimum": 1,
+                            "maximum": 100,
+                            "title": "Times",
+                            "description": "How many times to run each task.",
+                        },
                     },
                     "additionalProperties": False,
                 },

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/actions/util.py

@@ -11,6 +11,7 @@ from concurrent import futures
 from functools import reduce
 
 from requests.exceptions import HTTPError
+from taskcluster.exceptions import TaskclusterRestFailure
 
 from taskgraph import create
 from taskgraph.decision import read_artifact, rename_artifact, write_artifact
@@ -36,7 +37,7 @@ def fetch_graph_and_labels(parameters, graph_config, task_group_id=None):
     try:
         # Look up the decision_task id in the index
         decision_task_id = find_decision_task(parameters, graph_config)
-    except KeyError:
+    except (KeyError, TaskclusterRestFailure):
         if not task_group_id:
             raise
         # Not found (e.g. from github-pull-request), fall back to the task group id.

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/config.py

@@ -8,7 +8,6 @@ import os
 import sys
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Dict
 
 from voluptuous import ALLOW_EXTRA, All, Any, Extra, Length, Optional, Required
 
@@ -113,7 +112,7 @@ graph_config_schema = Schema(
 
 @dataclass(frozen=True, eq=False)
 class GraphConfig:
-    _config: Dict
+    _config: dict
     root_dir: str
 
     _PATH_MODIFIED = False

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/decision.py

@@ -21,7 +21,7 @@ from taskgraph.taskgraph import TaskGraph
 from taskgraph.util import json
 from taskgraph.util.python_path import find_object
 from taskgraph.util.schema import Schema, validate_schema
-from taskgraph.util.vcs import Repository, get_repository
+from taskgraph.util.vcs import get_repository
 from taskgraph.util.yaml import load_yaml
 
 logger = logging.getLogger(__name__)
@@ -192,25 +192,10 @@ def get_decision_parameters(graph_config, options):
     except UnicodeDecodeError:
         commit_message = ""
 
-    parameters["base_ref"] = _determine_more_accurate_base_ref(
-        repo,
-        candidate_base_ref=options.get("base_ref"),
-        head_ref=options.get("head_ref"),
-        base_rev=options.get("base_rev"),
-    )
-
-    parameters["base_rev"] = _determine_more_accurate_base_rev(
-        repo,
-        base_ref=parameters["base_ref"],
-        candidate_base_rev=options.get("base_rev"),
-        head_rev=options.get("head_rev"),
-        env_prefix=_get_env_prefix(graph_config),
-    )
-
     # Define default filter list, as most configurations shouldn't need
     # custom filters.
     parameters["files_changed"] = repo.get_changed_files(
-        rev=parameters["head_rev"], base_rev=parameters["base_rev"]
+        rev=parameters["head_rev"], base=parameters["base_rev"]
     )
     parameters["filters"] = [
         "target_tasks_method",
@@ -284,68 +269,6 @@ def get_decision_parameters(graph_config, options):
     return result
 
 
-def _determine_more_accurate_base_ref(repo, candidate_base_ref, head_ref, base_rev):
-    base_ref = candidate_base_ref
-
-    if not candidate_base_ref:
-        base_ref = repo.default_branch
-    elif candidate_base_ref == head_ref and base_rev == Repository.NULL_REVISION:
-        logger.info(
-            "base_ref and head_ref are identical but base_rev equals the null revision. "
-            "This is a new branch but Github didn't identify its actual base."
-        )
-        base_ref = repo.default_branch
-
-    if base_ref != candidate_base_ref:
-        logger.info(
-            f'base_ref has been reset from "{candidate_base_ref}" to "{base_ref}".'
-        )
-
-    return base_ref
-
-
-def _determine_more_accurate_base_rev(
-    repo, base_ref, candidate_base_rev, head_rev, env_prefix
-):
-    if not candidate_base_rev:
-        logger.info("base_rev is not set.")
-        base_ref_or_rev = base_ref
-    elif candidate_base_rev == Repository.NULL_REVISION:
-        logger.info("base_rev equals the null revision. This branch is a new one.")
-        base_ref_or_rev = base_ref
-    elif not repo.does_revision_exist_locally(candidate_base_rev):
-        logger.warning(
-            "base_rev does not exist locally. It is likely because the branch was force-pushed. "
-            "taskgraph is not able to assess how many commits were changed and assumes it is only "
-            f"the last one. Please set the {env_prefix.upper()}_BASE_REV environment variable "
-            "in the decision task and provide `--base-rev` to taskgraph."
-        )
-        base_ref_or_rev = base_ref
-    else:
-        base_ref_or_rev = candidate_base_rev
-
-    if base_ref_or_rev == base_ref:
-        logger.info(
-            f'Using base_ref "{base_ref}" to determine latest common revision...'
-        )
-
-    base_rev = repo.find_latest_common_revision(base_ref_or_rev, head_rev)
-    if base_rev != candidate_base_rev:
-        if base_ref_or_rev == candidate_base_rev:
-            logger.info("base_rev is not an ancestor of head_rev.")
-
-        logger.info(
-            f'base_rev has been reset from "{candidate_base_rev}" to "{base_rev}".'
-        )
-
-    return base_rev
-
-
-def _get_env_prefix(graph_config):
-    repo_keys = list(graph_config["taskgraph"].get("repositories", {}).keys())
-    return repo_keys[0] if repo_keys else ""
-
-
 def set_try_config(parameters, task_config_file):
     if os.path.isfile(task_config_file):
         logger.info(f"using try tasks from {task_config_file}")

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/docker.py

@@ -11,10 +11,11 @@ import subprocess
 import sys
 import tarfile
 import tempfile
+from collections.abc import Generator
 from io import BytesIO
 from pathlib import Path
 from textwrap import dedent
-from typing import Any, Dict, Generator, List, Optional, Union
+from typing import Any, Optional, Union
 
 from taskcluster.exceptions import TaskclusterRestFailure
 
@@ -246,7 +247,7 @@ def build_image(
 
 def load_image(
     url: str, imageName: Optional[str] = None, imageTag: Optional[str] = None
-) -> Dict[str, str]:
+) -> dict[str, str]:
     """Load docker image from URL as imageName:tag.
 
     Downloads a zstd-compressed docker image tarball from the given URL and
@@ -286,7 +287,7 @@ def load_image(
         else:
             imageTag = "latest"
 
-    info: Dict[str, str] = {}
+    info: dict[str, str] = {}
 
     def download_and_modify_image() -> Generator[bytes, None, None]:
         # This function downloads and edits the downloaded tar file on the fly.
@@ -377,14 +378,14 @@ def load_image(
     return info
 
 
-def _index(l: List, s: str) -> Optional[int]:
+def _index(l: list, s: str) -> Optional[int]:
     try:
         return l.index(s)
     except ValueError:
         pass
 
 
-def _resolve_image(image: Union[str, Dict[str, str]], graph_config: GraphConfig) -> str:
+def _resolve_image(image: Union[str, dict[str, str]], graph_config: GraphConfig) -> str:
     image_task_id = None
 
     # Standard case, image comes from the task definition.
@@ -419,12 +420,12 @@ def _resolve_image(image: Union[str, Dict[str, str]], graph_config: GraphConfig)
 
 def load_task(
     graph_config: GraphConfig,
-    task: Union[str, Dict[str, Any]],
+    task: Union[str, dict[str, Any]],
     remove: bool = True,
     user: Optional[str] = None,
     custom_image: Optional[str] = None,
     interactive: Optional[bool] = False,
-    volumes: Optional[Dict[str, str]] = None,
+    volumes: Optional[dict[str, str]] = None,
 ) -> int:
     """Load and run a task interactively in a Docker container.
 

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/generator.py

@@ -14,7 +14,7 @@ from concurrent.futures import (
     wait,
 )
 from dataclasses import dataclass
-from typing import Callable, Dict, Optional, Union
+from typing import Callable, Optional, Union
 
 from . import filter_tasks
 from .config import GraphConfig, load_graph_config
@@ -42,7 +42,7 @@ class KindNotFound(Exception):
 class Kind:
     name: str
     path: str
-    config: Dict
+    config: dict
     graph_config: GraphConfig
 
     def _get_loader(self) -> Callable:

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/graph.py

@@ -4,12 +4,19 @@
 
 
 import collections
+import functools
 from dataclasses import dataclass
-from typing import FrozenSet
+
+from .util.readonlydict import ReadOnlyDict
 
 
 @dataclass(frozen=True)
-class Graph:
+class _Graph:
+    nodes: frozenset
+    edges: frozenset
+
+
+class Graph(_Graph):
     """Generic representation of a directed acyclic graph with labeled edges
     connecting the nodes. Graph operations are implemented in a functional
     manner, so the data structure is immutable.
@@ -24,8 +31,8 @@ class Graph:
     node `left` to node `right`..
     """
 
-    nodes: FrozenSet
-    edges: FrozenSet
+    def __init__(self, nodes, edges):
+        super().__init__(frozenset(nodes), frozenset(edges))
 
     def transitive_closure(self, nodes, reverse=False):
         """Return the transitive closure of <nodes>: the graph containing all
@@ -68,23 +75,28 @@ class Graph:
             add_nodes = {(left if reverse else right) for (left, right, _) in add_edges}
             new_nodes = nodes | add_nodes
             new_edges = edges | add_edges
-        return Graph(new_nodes, new_edges)  # type: ignore
+        return Graph(new_nodes, new_edges)
 
     def _visit(self, reverse):
-        queue = collections.deque(sorted(self.nodes))
-        links_by_node = self.reverse_links_dict() if reverse else self.links_dict()
-        seen = set()
+        forward_links, reverse_links = self.links_and_reverse_links_dict()
+
+        dependencies = reverse_links if reverse else forward_links
+        dependents = forward_links if reverse else reverse_links
+
+        indegree = {node: len(dependencies[node]) for node in self.nodes}
+
+        queue = collections.deque(
+            node for node, degree in indegree.items() if degree == 0
+        )
+
         while queue:
             node = queue.popleft()
-            if node in seen:
-                continue
-            links = links_by_node[node]
-            if all((n in seen) for n in links):
-                seen.add(node)
-                yield node
-            else:
-                queue.extend(n for n in links if n not in seen)
-                queue.append(node)
+            yield node
+
+            for dependent in dependents[node]:
+                indegree[dependent] -= 1
+                if indegree[dependent] == 0:
+                    queue.append(dependent)
 
     def visit_postorder(self):
         """
@@ -103,6 +115,30 @@ class Graph:
         """
         return self._visit(True)
 
+    @functools.cache
+    def links_and_reverse_links_dict(self):
+        """
+        Return both links and reverse_links dictionaries.
+        Returns a (forward_links, reverse_links) tuple where forward_links maps
+        each node to the set of nodes it links to, and reverse_links maps each
+        node to the set of nodes linking to it.
+
+        Because the return value is cached, this function returns a pair of
+        `ReadOnlyDict` instead of defaultdicts like its `links_dict` and
+        `reverse_links_dict` counterparts to avoid consumers modifying the
+        cached value by mistake.
+        """
+        forward = {node: set() for node in self.nodes}
+        reverse = {node: set() for node in self.nodes}
+        for left, right, _ in self.edges:
+            forward[left].add(right)
+            reverse[right].add(left)
+
+        return (
+            ReadOnlyDict({key: frozenset(value) for key, value in forward.items()}),
+            ReadOnlyDict({key: frozenset(value) for key, value in reverse.items()}),
+        )
+
     def links_dict(self):
         """
         Return a dictionary mapping each node to a set of the nodes it links to

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/main.py

@@ -16,7 +16,7 @@ from collections import namedtuple
 from concurrent.futures import ProcessPoolExecutor, as_completed
 from pathlib import Path
 from textwrap import dedent
-from typing import Any, List
+from typing import Any
 from urllib.parse import urlparse
 
 import appdirs
@@ -412,7 +412,7 @@ def show_taskgraph(options):
     overrides = {
         "target-kinds": options.get("target_kinds"),
     }
-    parameters: List[Any[str, Parameters]] = options.pop("parameters")
+    parameters: list[Any[str, Parameters]] = options.pop("parameters")
     if not parameters:
         parameters = [
             parameters_loader(None, strict=False, overrides=overrides)

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/morph.py

@@ -51,7 +51,7 @@ def amend_taskgraph(taskgraph, label_to_taskid, to_add):
         for depname, dep in task.dependencies.items():
             new_edges.add((task.task_id, dep, depname))
 
-    taskgraph = TaskGraph(new_tasks, Graph(set(new_tasks), new_edges))  # type: ignore
+    taskgraph = TaskGraph(new_tasks, Graph(frozenset(new_tasks), new_edges))
     return taskgraph, label_to_taskid
 
 

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/optimize/base.py

@@ -15,7 +15,6 @@ import datetime
 import logging
 from abc import ABCMeta, abstractmethod
 from collections import defaultdict
-from typing import Dict, Set
 
 from slugid import nice as slugid
 
@@ -298,8 +297,9 @@ def replace_tasks(
 
     opt_counts = defaultdict(int)
     replaced = set()
-    dependents_of = target_task_graph.graph.reverse_links_dict()
-    dependencies_of = target_task_graph.graph.links_dict()
+    dependencies_of, dependents_of = (
+        target_task_graph.graph.links_and_reverse_links_dict()
+    )
 
     for label in target_task_graph.graph.visit_postorder():
         logger.debug(f"replace_tasks: {label}")
@@ -367,9 +367,9 @@ def replace_tasks(
 
 def get_subgraph(
     target_task_graph: TaskGraph,
-    removed_tasks: Set[str],
-    replaced_tasks: Set[str],
-    label_to_taskid: Dict[str, str],
+    removed_tasks: set[str],
+    replaced_tasks: set[str],
+    label_to_taskid: dict[str, str],
     decision_task_id: str,
 ):
     """
@@ -452,7 +452,9 @@ def get_subgraph(
         if left in tasks_by_taskid and right in tasks_by_taskid
     }
 
-    return TaskGraph(tasks_by_taskid, Graph(set(tasks_by_taskid), edges_by_taskid))  # type: ignore
+    return TaskGraph(
+        tasks_by_taskid, Graph(frozenset(tasks_by_taskid), edges_by_taskid)
+    )
 
 
 @register_strategy("never")

{taskcluster_taskgraph-16.2.0 → taskcluster_taskgraph-17.0.0}/src/taskgraph/optimize/strategies.py

@@ -1,6 +1,8 @@
 import logging
 from datetime import datetime
 
+from taskcluster.exceptions import TaskclusterRestFailure
+
 from taskgraph.optimize.base import OptimizationStrategy, register_strategy
 from taskgraph.util.path import match as match_path
 from taskgraph.util.taskcluster import find_task_id, status_task
@@ -64,7 +66,7 @@ class IndexSearch(OptimizationStrategy):
                     continue
 
                 return task_id
-            except KeyError:
+            except (KeyError, TaskclusterRestFailure):
                 # go on to the next index path
                 pass