| 321 | captured: Dict[str, Any] = {} |
| 322 | |
| 323 | async def fake_request_json(_, method: str, path: str, *, json: Any = None, params: Any = None) -> Any: |
| 324 | captured.update({"method": method, "path": path, "json": json}) |
| 325 | count = len(json["rollouts"]) if json and "rollouts" in json else 0 # type: ignore[index] |
| 326 | return [{"rollout_id": f"bulk-{idx}", "input": {"idx": idx}, "start_time": float(idx)} for idx in range(count)] |
| 327 | |
| 328 | monkeypatch.setattr(LightningStoreClient, "_request_json", fake_request_json, raising=False) # type: ignore |
| 329 | |