Unverified 提交 bc484579 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Handle non-TTY `wandb.errors.UsageError` (#5839)

* `try: except (..., wandb.errors.UsageError)` * bug fix
上级 a4207a20
...@@ -24,7 +24,10 @@ try: ...@@ -24,7 +24,10 @@ try:
assert hasattr(wandb, '__version__') # verify package import not local dir assert hasattr(wandb, '__version__') # verify package import not local dir
if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2') and RANK in [0, -1]: if pkg.parse_version(wandb.__version__) >= pkg.parse_version('0.12.2') and RANK in [0, -1]:
wandb_login_success = wandb.login(timeout=30) try:
wandb_login_success = wandb.login(timeout=30)
except wandb.errors.UsageError: # known non-TTY terminal issue
wandb_login_success = False
if not wandb_login_success: if not wandb_login_success:
wandb = None wandb = None
except (ImportError, AssertionError): except (ImportError, AssertionError):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论