diff --git a/utils/__init__.py b/utils/__init__.py index bccac42..4687889 100644 --- a/utils/__init__.py +++ b/utils/__init__.py @@ -61,6 +61,7 @@ def notebook_init(verbose=True): import psutil + os.system('pip uninstall -y wandb') if is_colab(): shutil.rmtree('/content/sample_data', ignore_errors=True) # remove colab /sample_data directory