Use spot machines on GCP

populationgenomics · Jun 1, 2022 · 13c377b · 13c377b
1 parent 6b70db7
commit 13c377b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/batch/batch/cloud/gcp/driver/create_instance.py b/batch/batch/cloud/gcp/driver/create_instance.py
@@ -96,7 +96,13 @@ def create_vm_config(
                 'accessConfigs': [{'type': 'ONE_TO_ONE_NAT', 'name': 'external-nat'}],
             }
         ],
-        'scheduling': {'automaticRestart': False, 'onHostMaintenance': "TERMINATE", 'preemptible': preemptible},
+        'scheduling': {
+            'automaticRestart': False,
+            'onHostMaintenance': 'TERMINATE',
+            'provisioningModel': 'SPOT' if preemptible else 'STANDARD',
+            'instanceTerminationAction': 'DELETE',
+            'preemptible': preemptible
+        },
         'serviceAccounts': [
             {
                 'email': f'batch2-agent@{project}.iam.gserviceaccount.com',