Skip to content

Commit

Permalink
hack to work around mcli bug
Browse files Browse the repository at this point in the history
  • Loading branch information
epwalsh committed Apr 12, 2024
1 parent 1eafd6b commit 5660857
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion scripts/mcli/manage_run.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@

import argparse
import sys
import time
from concurrent.futures import as_completed
from typing import List, Optional, Set

Expand Down Expand Up @@ -76,8 +77,11 @@ def wait_on_runs(runs: List[Run], timeout: int = _DEFAULT_TIMEOUT) -> List[Run]:
Wait on a list of runs to reach 'COMPLETED' status (or a failure of some kind).
"""
futures = []
for run in runs:
for i, run in enumerate(runs):
futures.append(mcli.api.runs.wait_for_run_status(run, RunStatus.COMPLETED, future=True))
if i == 0:
# HACK: this works around a bug in `mcli`.
time.sleep(0.05)

results = []
for future in track(
Expand Down

0 comments on commit 5660857

Please sign in to comment.