Uninstall `wandb` from notebook environments (#3247)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
single_channel
Glenn Jocher 1 year ago committed by GitHub
parent 064d48eff9
commit 4c2033d7c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -20,8 +20,8 @@ import torch
from matplotlib import font_manager from matplotlib import font_manager
from ultralytics.yolo.utils import (AUTOINSTALL, LOGGER, ONLINE, ROOT, USER_CONFIG_DIR, TryExcept, clean_url, colorstr, from ultralytics.yolo.utils import (AUTOINSTALL, LOGGER, ONLINE, ROOT, USER_CONFIG_DIR, TryExcept, clean_url, colorstr,
downloads, emojis, is_colab, is_docker, is_kaggle, is_online, is_pip_package, downloads, emojis, is_colab, is_docker, is_jupyter, is_kaggle, is_online,
url2file) is_pip_package, url2file)
def is_ascii(s) -> bool: def is_ascii(s) -> bool:
@ -325,6 +325,9 @@ def check_yolo(verbose=True, device=''):
"""Return a human-readable YOLO software and hardware summary.""" """Return a human-readable YOLO software and hardware summary."""
from ultralytics.yolo.utils.torch_utils import select_device from ultralytics.yolo.utils.torch_utils import select_device
if is_jupyter():
if check_requirements('wandb', install=False):
os.system('pip uninstall -y wandb') # uninstall wandb: unwanted account creation prompt with infinite hang
if is_colab(): if is_colab():
shutil.rmtree('sample_data', ignore_errors=True) # remove colab /sample_data directory shutil.rmtree('sample_data', ignore_errors=True) # remove colab /sample_data directory

Loading…
Cancel
Save