diff --git a/.github/workflows/alarm.yml b/.github/workflows/alarm.yml index 050c5e8020..4ba8ec25d4 100644 --- a/.github/workflows/alarm.yml +++ b/.github/workflows/alarm.yml @@ -31,9 +31,7 @@ jobs: - name: Install deps run: | - # Pin wandb<0.25: 0.25.x changed api.runs() to no longer populate - # run.config in batch queries, breaking alarm.py's config["revision"] lookup. - python -m pip install --quiet --upgrade 'wandb<0.25' frozendict + python -m pip install --quiet --upgrade wandb frozendict - name: Download speed artifacts from triggering run id: dl_speed diff --git a/.github/workflows/scripts/alarm.py b/.github/workflows/scripts/alarm.py index 3fdd8ec0e1..e11c099080 100644 --- a/.github/workflows/scripts/alarm.py +++ b/.github/workflows/scripts/alarm.py @@ -285,7 +285,7 @@ def fetch_wandb_data( wandb_parser: WandbParser, ) -> dict[str, dict[frozendict[str, str], dict[str, float | int]]]: api = wandb.Api() - runs_iter: Iterable[Run] = api.runs(f"{self.wandb_entity}/{wandb_parser.project}", order="-created_at") + runs_iter: Iterable[Run] = api.runs(f"{self.wandb_entity}/{wandb_parser.project}", order="-created_at", lazy=False) commit_hashes = set() records_by_commit_hash: dict[str, dict[frozendict[str, str], dict[str, float | int]]] = defaultdict(