From cfe0fdb3dded2b3cd3a24db30c3ff4796cdf742a Mon Sep 17 00:00:00 2001
From: Google Colaboratory Team <colaboratory-team@google.com>
Date: Wed, 15 May 2024 14:11:37 -0700
Subject: [PATCH] No public description

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

diff --git a/google/colab/drive.py b/google/colab/drive.py
index 13a2d054..26f49d24 100644
--- a/google/colab/drive.py
+++ b/google/colab/drive.py
@@ -131,7 +131,9 @@ def _mount(
   )
   if ephemeral:
     _message.blocking_request(
-        'request_auth', request={'authType': 'dfs_ephemeral'}, timeout_sec=None
+        'request_auth',
+        request={'authType': 'dfs_ephemeral'},
+        timeout_sec=int(timeout_ms / 1000),
     )
 
   mountpoint = _os.path.expanduser(mountpoint)