diff --git a/allgo/__init__.py b/allgo/__init__.py index b8e347527d5a73a8d000654cfd08bfb6b23b40dc..f5ac6fda17648baa988d7effcba8509796c6d538 100644 --- a/allgo/__init__.py +++ b/allgo/__init__.py @@ -68,11 +68,12 @@ class App: r.raise_for_status() results = r.json() status = results['status'] - if status == 'in progress': # 'in progress', 'done', 'error', 'timeout' - log.info("wait for job %s", jobid) + if status in ['created', 'waiting', 'running', 'in progress']: + log.info("wait for job %s in status", jobid, status) time.sleep(2) else: break + if status != 'done': raise Exception('Job %s failed with status %s', (jobid, status))