From 43024a2d13c6de507238c3721b006c7efa35a364 Mon Sep 17 00:00:00 2001
From: Google Colaboratory Team <colaboratory-team@google.com>
Date: Tue, 10 Sep 2024 14:43:36 -0700
Subject: [PATCH] No public description

PiperOrigin-RevId: 673112582
---
 google/colab/drive.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/google/colab/drive.py b/google/colab/drive.py
index 01ecdf5f..f070b8ab 100644
--- a/google/colab/drive.py
+++ b/google/colab/drive.py
@@ -124,6 +124,10 @@ def _mount(
   if ' ' in mountpoint:
     raise ValueError('Mountpoint must not contain a space.')
 
+  if _os.environ.get('VERTEX_PRODUCT') == 'COLAB_ENTERPRISE':
+    raise NotImplementedError(
+        'google.colab.drive.mount is not supported in Colab Enterprise.'
+    )
   metadata_server_addr = (
       _os.environ['TBE_EPHEM_CREDS_ADDR']
       if ephemeral