fix image selection

EthDevOps · Apr 15, 2024 · c94b94e · c94b94e
1 parent f3a4861
commit c94b94e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/CloudController.cs b/CloudController.cs
@@ -51,10 +51,16 @@ public async Task<string> CreateNewRunner(string arch, string size, string runne
         string name = $"ghr-{NameGenerator.Identifiers.Get(separator: "-")}".ToLower();
 
         _logger.LogInformation($"Creating VM {name} from image {imageName} of size {size}");
+
+        string htzArch = "x86";
+        if (arch == "arm64")
+        {
+            htzArch = "arm";
+        }
 
         // Grab image
         var images = await _client.Image.Get();
-        long? imageId =  images.FirstOrDefault(x => x.Description == imageName)?.Id;
+        long? imageId =  images.FirstOrDefault(x => x.Description == imageName && x.Architecture == htzArch)?.Id;
 
         if (!imageId.HasValue)
         {