Fix `is_colab()` (#9994)

@AyushExel @kalenmike Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-01 12:02:05 +01:00 · 2022-11-01 12:02:05 +01:00 · 49156eb1d1
parent a83d2a5013
commit 49156eb1d1
1 changed files with 1 additions and 1 deletions
--- a/utils/general.py
+++ b/utils/general.py
@ -72,7 +72,7 @@ def is_chinese(s='人工智能'):

 def is_colab():
    # Is environment a Google Colab instance?
-    return 'COLAB_GPU' in os.environ
+    return 'google.colab' in sys.modules


 def is_notebook():