diff --git a/ultralytics/yolo/utils/checks.py b/ultralytics/yolo/utils/checks.py index eb30da1..a2d571f 100644 --- a/ultralytics/yolo/utils/checks.py +++ b/ultralytics/yolo/utils/checks.py @@ -20,8 +20,8 @@ import torch from matplotlib import font_manager 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, - url2file) + downloads, emojis, is_colab, is_docker, is_jupyter, is_kaggle, is_online, + is_pip_package, url2file) def is_ascii(s) -> bool: @@ -325,8 +325,11 @@ def check_yolo(verbose=True, device=''): """Return a human-readable YOLO software and hardware summary.""" from ultralytics.yolo.utils.torch_utils import select_device - if is_colab(): - shutil.rmtree('sample_data', ignore_errors=True) # remove colab /sample_data directory + 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(): + shutil.rmtree('sample_data', ignore_errors=True) # remove colab /sample_data directory if verbose: # System info