Fix tests

dask · Sep 19, 2024 · 98caf0b · 98caf0b
1 parent 1d4546b
commit 98caf0b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dask_cloudprovider/gcp/instances.py b/dask_cloudprovider/gcp/instances.py
@@ -19,7 +19,6 @@
 
 try:
     import googleapiclient.discovery
-    from google.auth import default as google_auth_default
     from googleapiclient.errors import HttpError
 except ImportError as e:
     msg = (
@@ -671,8 +670,9 @@ def refresh_client(self):
                 )
             )
         else:
+            import google.auth
             # Obtain Application Default Credentials
-            credentials, _ = google_auth_default()
+            credentials, _ = google.auth.default()
 
         # Use the credentials to build a service client
         return googleapiclient.discovery.build(