Test if notebook is running on Google Colab

Try importing google.colab

try:
  import google.colab
  IN_COLAB = True
except:
  IN_COLAB = False

Or just check if it’s in sys.modules

import sys
IN_COLAB = 'google.colab' in sys.modules