Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Uninstall wandb from notebook environments #11730

Merged
merged 3 commits into from
Jun 18, 2023
Merged

Uninstall wandb from notebook environments #11730

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
e9b92e2
Uninstall `wandb` from notebook environments
glenn-jocher Jun 18, 2023
1c7756a
fix import
glenn-jocher Jun 18, 2023
6dbb70e
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 18, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion utils/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down