Rebuild agent on original pooled runner
This commit is contained in:
@@ -7,7 +7,7 @@ from typing import Any
|
||||
|
||||
from common import append_jsonl, parse_datetime, read_jsonl, utc_now, write_jsonl
|
||||
from history_stats import classify_failure, is_failure, is_success
|
||||
from modelhub_client import ModelHubClient
|
||||
from modelhub_client import ModelHubClient, ModelHubClientPool
|
||||
|
||||
|
||||
DEFAULT_OUTCOMES_PATH = Path("outcomes/submissions.jsonl")
|
||||
@@ -72,11 +72,14 @@ class OutcomeTracker:
|
||||
self._by_model_gpu[(model_id, target_gpu)].append(record)
|
||||
append_jsonl(self.path, record)
|
||||
|
||||
def sync_from_api(self, client: ModelHubClient) -> int:
|
||||
def sync_from_api(self, client: ModelHubClient | ModelHubClientPool) -> int:
|
||||
try:
|
||||
begin = self._last_sync_time
|
||||
end = utc_now()
|
||||
# Use fanout for pools to sync outcomes across all accounts
|
||||
list_kwargs: dict[str, Any] = {"begin_time": begin, "end_time": end, "page_size": 100, "only_mine": True}
|
||||
if isinstance(client, ModelHubClientPool):
|
||||
list_kwargs["_fanout_all"] = True
|
||||
tasks = client.list_tasks(**list_kwargs)
|
||||
except Exception:
|
||||
return 0
|
||||
|
||||
Reference in New Issue
Block a user