Skip to content

Commit 75993df

Browse files
committed
Add missing request builder
1 parent f3a7ca2 commit 75993df

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

dask_cloudprovider/gcp/instances.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -626,9 +626,11 @@ def __init__(
626626
"gpu_instance": self.gpu_instance,
627627
"bootstrap": self.bootstrap,
628628
"auto_shutdown": self.auto_shutdown,
629-
"preemptible": preemptible
630-
if preemptible is not None
631-
else self.config.get("preemptible"),
629+
"preemptible": (
630+
preemptible
631+
if preemptible is not None
632+
else self.config.get("preemptible")
633+
),
632634
"instance_labels": instance_labels or self.config.get("instance_labels"),
633635
"service_account": service_account or self.config.get("service_account"),
634636
}
@@ -652,7 +654,12 @@ def refresh_client(self):
652654
credentials, _ = google.auth.default()
653655

654656
# Use the credentials to build a service client
655-
return googleapiclient.discovery.build("compute", "v1", credentials=credentials)
657+
return googleapiclient.discovery.build(
658+
"compute",
659+
"v1",
660+
credentials=credentials,
661+
requestBuilder=build_request(credentials),
662+
)
656663

657664
def instances(self):
658665
try:

0 commit comments

Comments
 (0)