Add optional scheduling

SpareCores · Jun 20, 2024 · 88fab0a · 88fab0a
1 parent a4fbdac
commit 88fab0a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/sc_runner/resources/gcp.py b/src/sc_runner/resources/gcp.py
@@ -37,8 +37,7 @@ def resources_gcp(
             instance_opts["metadata"]["ssh-keys"] = f"ubuntu:{public_key}"
         else:
             instance_opts["metadata"] = {"ssh-keys": f"ubuntu:{public_key}"}
-    gcp.compute.Instance(
-        instance,
+    instance_opts |= dict(
         machine_type=instance,
         zone=zone,
         boot_disk=gcp.compute.InstanceBootDiskArgs(
@@ -51,7 +50,11 @@ def resources_gcp(
                 access_configs=[gcp.compute.InstanceNetworkInterfaceAccessConfigArgs()]
             )
         ],
-        scheduling=gcp.compute.InstanceSchedulingArgs(**scheduling_opts),
+    )
+    if scheduling_opts:
+        instance_opts["scheduling"] = gcp.compute.InstanceSchedulingArgs(**scheduling_opts)
+    gcp.compute.Instance(
+        instance,
         **instance_opts,
         opts=pulumi.ResourceOptions(provider=provider),
     )