From 28fe2006db35432d58e9144b17a42873eda4e15d Mon Sep 17 00:00:00 2001
From: Thomas Carmet <8408330+tcarmet@users.noreply.github.com>
Date: Mon, 22 Jul 2024 14:40:29 -0700
Subject: [PATCH] Support for referrencing image name on gcp backend (#654)

---
 runner_manager/backend/gcloud.py | 5 +++++
 runner_manager/models/backend.py | 1 +
 2 files changed, 6 insertions(+)

diff --git a/runner_manager/backend/gcloud.py b/runner_manager/backend/gcloud.py
index 72dffe46..5367ff89 100644
--- a/runner_manager/backend/gcloud.py
+++ b/runner_manager/backend/gcloud.py
@@ -73,6 +73,11 @@ def wait_for_operation(
 
     @property
     def image(self) -> Image:
+        if self.instance_config.image is not None:
+            return self.image_client.get(
+                project=self.instance_config.image_project,
+                image=self.instance_config.image,
+            )
         return self.image_client.get_from_family(
             project=self.instance_config.image_project,
             family=self.instance_config.image_family,
diff --git a/runner_manager/models/backend.py b/runner_manager/models/backend.py
index 5cd411df..599d7ac2 100644
--- a/runner_manager/models/backend.py
+++ b/runner_manager/models/backend.py
@@ -108,6 +108,7 @@ class GCPConfig(BackendConfig):
 class GCPInstanceConfig(InstanceConfig):
     image_family: str = "ubuntu-2004-lts"
     image_project: str = "ubuntu-os-cloud"
+    image: Optional[str] = None
     machine_type: str = "e2-small"
     subnetwork: str = "default"
     enable_nested_virtualization: bool = True