Skip to content

Commit

Permalink
Sleeping messages should be info, not error (#198)
Browse files Browse the repository at this point in the history
wukevin authored Dec 1, 2024
1 parent fc030e4 commit e80bb3a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions chai_lab/data/dataset/msas/colabfold.py
Original file line number Diff line number Diff line change
@@ -194,7 +194,7 @@ def download(ID, path):
out = submit(seqs_unique, mode, N)
while out["status"] in ["UNKNOWN", "RATELIMIT"]:
sleep_time = 5 + random.randint(0, 5)
logger.error(f"Sleeping for {sleep_time}s. Reason: {out['status']}")
logger.info(f"Sleeping for {sleep_time}s. Reason: {out['status']}")
# resubmit
time.sleep(sleep_time)
out = submit(seqs_unique, mode, N)
@@ -214,7 +214,7 @@ def download(ID, path):
pbar.set_description(out["status"])
while out["status"] in ["UNKNOWN", "RUNNING", "PENDING"]:
t = 5 + random.randint(0, 5)
logger.error(f"Sleeping for {t}s. Reason: {out['status']}")
logger.info(f"Sleeping for {t}s. Reason: {out['status']}")
time.sleep(t)
out = status(ID)
pbar.set_description(out["status"])

0 comments on commit e80bb3a

Please sign in to comment.