Skip to content

Commit b6261f1

Browse files
committed
Add missing request builder
1 parent 3c5a3f4 commit b6261f1

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
@@ -629,9 +629,11 @@ def __init__(
629629
"gpu_instance": self.gpu_instance,
630630
"bootstrap": self.bootstrap,
631631
"auto_shutdown": self.auto_shutdown,
632-
"preemptible": preemptible
633-
if preemptible is not None
634-
else self.config.get("preemptible"),
632+
"preemptible": (
633+
preemptible
634+
if preemptible is not None
635+
else self.config.get("preemptible")
636+
),
635637
"instance_labels": instance_labels or self.config.get("instance_labels"),
636638
"service_account": service_account or self.config.get("service_account"),
637639
}
@@ -671,7 +673,12 @@ def refresh_client(self):
671673
credentials, _ = google.auth.default()
672674

673675
# Use the credentials to build a service client
674-
return googleapiclient.discovery.build("compute", "v1", credentials=credentials)
676+
return googleapiclient.discovery.build(
677+
"compute",
678+
"v1",
679+
credentials=credentials,
680+
requestBuilder=build_request(credentials),
681+
)
675682

676683
def instances(self):
677684
try:

0 commit comments

Comments
 (0)