add graceful termination to nextflow runtime task

latchbio · Oct 11, 2024 · 8c4b134 · 8c4b134
1 parent 122f91c
commit 8c4b134
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/latch/resources/tasks.py b/latch/resources/tasks.py
@@ -606,6 +606,8 @@ def nextflow_runtime_task(cpu: int, memory: int, storage_gib: int = 50):
     task_config = _custom_task_config(cpu, memory, storage_gib)
 
     task_config.pod_spec.automount_service_account_token = True
+    # rahul: give flyte enough time to process the task after completion
+    task_config.pod_spec.termination_grace_period_seconds = 120
 
     assert len(task_config.pod_spec.containers) == 1
     task_config.pod_spec.containers[0].volume_mounts = [