diff --git a/benchmarks/ax/search_probe.py b/benchmarks/ax/search_probe.py index e9d5e92..5558b7d 100644 --- a/benchmarks/ax/search_probe.py +++ b/benchmarks/ax/search_probe.py @@ -30,12 +30,12 @@ def validate_result(data: bytes, output: Path) -> None: def probe_workload(image: str, deadline: float = 900) -> Workload: validate_image(image) - return Workload(image, frozenset({"RESULT_URL"}), deadline, validate_result) + return Workload(image, frozenset({"RESULT_URL"}), deadline, validate_result, kind="search-probe") def probe_item(image: str) -> Item: return Item( - "search-probe", + "v1", {}, "result.json", ("python3", "-m", "openultrasast.search.probe"), diff --git a/benchmarks/ax/search_verify.py b/benchmarks/ax/search_verify.py index 082ad00..791917d 100644 --- a/benchmarks/ax/search_verify.py +++ b/benchmarks/ax/search_verify.py @@ -66,7 +66,7 @@ class AXSideDispatcher: raise ValueError("verification accepts checkout and declarative inputs only") # The trusted host selects an executor for this side. Build/package work # finishes there before any verifier worker slot is acquired. - workload = Workload(self.image, frozenset({"VERIFY_INPUT_URL", "RESULT_URL"}), timeout_seconds * 6 + 60, validate_result) + workload = Workload(self.image, frozenset({"VERIFY_INPUT_URL", "RESULT_URL"}), timeout_seconds * 6 + 60, validate_result, kind="search-verify") lane = self.lane or AXLane(self.args, workload) spec = {"demo": demo, "family": family, "timeout_seconds": timeout_seconds} try: @@ -79,7 +79,7 @@ class AXSideDispatcher: return SideRecord("could_not_build", (), 0, str(exc), isolation_mode="task-boundary") records = [] for _ in range(3): - identity = "search-verify-" + uuid.uuid4().hex + identity = uuid.uuid4().hex item = Item( identity, {"VERIFY_INPUT_URL": archive}, diff --git a/benchmarks/search/pilot_run.py b/benchmarks/search/pilot_run.py index d604ed5..17a9e73 100644 --- a/benchmarks/search/pilot_run.py +++ b/benchmarks/search/pilot_run.py @@ -501,7 +501,7 @@ def main(argv=None): try: store = open_store(args.board_memory) if args.board_memory else FileStore(root / "boards") if not args.dry_run: - lane = AXLane(args, Workload(args.image, frozenset({"VERIFY_INPUT_URL", "RESULT_URL"}), 900, validate_result)) + lane = AXLane(args, Workload(args.image, frozenset({"VERIFY_INPUT_URL", "RESULT_URL"}), 900, validate_result, kind="search-verify")) except Exception as exc: aborted_reason = exception_reason(exc) for index in indices if aborted_reason is None else []: