Uninstall wandb from notebook environments (ultralytics#11730)

* Uninstall `wandb` from notebook environments Due to undesired behavior in https://www.kaggle.com/code/ultralytics/yolov8/comments#2306977 Signed-off-by: Glenn Jocher <[email protected]> * fix import * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: Glenn Jocher <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
irajcode · Jul 20, 2023 · 4174db9 · 4174db9
1 parent 09a15b0
commit 4174db9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/utils/__init__.py b/utils/__init__.py
@@ -54,14 +54,17 @@ def notebook_init(verbose=True):
     import os
     import shutil
 
+    from ultralytics.yolo.utils.checks import check_requirements
+
     from utils.general import check_font, is_colab
     from utils.torch_utils import select_device  # imports
 
     check_font()
 
     import psutil
 
-    os.system('pip uninstall -y wandb')
+    if check_requirements('wandb', install=False):
+        os.system('pip uninstall -y wandb')  # eliminate unexpected account creation prompt with infinite hang
     if is_colab():
         shutil.rmtree('/content/sample_data', ignore_errors=True)  # remove colab /sample_data directory