Merge pull request #106 from trickest/hotfix/machine-config

Omit machine types unless they are > 0
trickest · Dec 20, 2023 · 099bd41 · 099bd41
2 parents fd8cef0 + 9d96ba8
commit 099bd41
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/cmd/execute/execute.go b/cmd/execute/execute.go
@@ -153,12 +153,21 @@ func parseMachineConfiguration(config string) (types.Machines, error) {
 			}
 		}
 
-		return types.Machines{
-			// sizes = [small, medium, large]
-			Small:  &sizes[0],
-			Medium: &sizes[1],
-			Large:  &sizes[2],
-		}, nil
+		var machines types.Machines
+
+		if sizes[0] != 0 {
+			machines.Small = &sizes[0]
+		}
+
+		if sizes[1] != 0 {
+			machines.Medium = &sizes[1]
+		}
+
+		if sizes[2] != 0 {
+			machines.Large = &sizes[2]
+		}
+
+		return machines, nil
 	}
 
 	// One type of machine