Skip to content

Commit 28fe200

Browse files
authored
Support for referrencing image name on gcp backend (#654)
1 parent 5ff35da commit 28fe200

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

runner_manager/backend/gcloud.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,11 @@ def wait_for_operation(
7373

7474
@property
7575
def image(self) -> Image:
76+
if self.instance_config.image is not None:
77+
return self.image_client.get(
78+
project=self.instance_config.image_project,
79+
image=self.instance_config.image,
80+
)
7681
return self.image_client.get_from_family(
7782
project=self.instance_config.image_project,
7883
family=self.instance_config.image_family,

runner_manager/models/backend.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,7 @@ class GCPConfig(BackendConfig):
108108
class GCPInstanceConfig(InstanceConfig):
109109
image_family: str = "ubuntu-2004-lts"
110110
image_project: str = "ubuntu-os-cloud"
111+
image: Optional[str] = None
111112
machine_type: str = "e2-small"
112113
subnetwork: str = "default"
113114
enable_nested_virtualization: bool = True

0 commit comments

Comments
 (0)