diff --git a/taskcluster/fxci_config_taskgraph/util/integration.py b/taskcluster/fxci_config_taskgraph/util/integration.py index 06bb775f..e5d770a9 100644 --- a/taskcluster/fxci_config_taskgraph/util/integration.py +++ b/taskcluster/fxci_config_taskgraph/util/integration.py @@ -7,8 +7,7 @@ import requests import taskcluster -from taskgraph.util.taskcluster import get_ancestors -from taskgraph.util.taskcluster import logging +from taskgraph.util.taskcluster import get_ancestors, logging FIREFOXCI_ROOT_URL = "https://firefox-ci-tc.services.mozilla.com" STAGING_ROOT_URL = "https://stage.taskcluster.nonprod.cloudops.mozgcp.net" @@ -21,7 +20,9 @@ def get_taskcluster_client(service: str): @cache -def find_tasks(decision_index_path: str, include_deps: bool = False) -> list[dict[str, Any]]: +def find_tasks( + decision_index_path: str, include_deps: bool = False +) -> list[dict[str, Any]]: """Find tasks targeted by the Decision task pointed to by `decision_index_path`.""" queue = get_taskcluster_client("queue") index = get_taskcluster_client("index") @@ -57,6 +58,7 @@ def find_tasks(decision_index_path: str, include_deps: bool = False) -> list[dic logging.info("including deps") # TODO: remove hack import os + orig = os.environ["TASKCLUSTER_ROOT_URL"] os.environ["TASKCLUSTER_ROOT_URL"] = FIREFOXCI_ROOT_URL logging.info(f"fetching ancestors for {task['task_id']}")