util.py

import os

def is_google_colab():
    if os.getenv("COLAB_RELEASE_TAG"):
        return True
    return False