wait for final_eval_task to complete b4 deleting pvcs

Signed-off-by: sallyom <[email protected]>
opendatahub-io · Oct 18, 2024 · 4697fb1 · 4697fb1
1 parent 9d45526
commit 4697fb1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/pipeline.py b/pipeline.py
@@ -377,13 +377,13 @@ def pipeline(
         )
 
         output_pvc_delete_task = DeletePVC(pvc_name=output_pvc_task.output)
-        output_pvc_delete_task.after(output_data_task)
+        output_pvc_delete_task.after(final_eval_task, output_data_task)
 
         sdg_pvc_delete_task = DeletePVC(pvc_name=sdg_input_pvc_task.output)
-        sdg_pvc_delete_task.after(output_data_task)
+        sdg_pvc_delete_task.after(final_eval_task, output_data_task)
 
         model_pvc_delete_task = DeletePVC(pvc_name=model_pvc_task.output)
-        model_pvc_delete_task.after(output_data_task)
+        model_pvc_delete_task.after(final_eval_task, output_data_task)
 
         return
 

diff --git a/pipeline.yaml b/pipeline.yaml
@@ -1549,6 +1549,7 @@ root:
         dependentTasks:
         - createpvc-3
         - pvc-to-model-op
+        - run-final-eval-op
         inputs:
           parameters:
             pvc_name:
@@ -1565,6 +1566,7 @@ root:
         dependentTasks:
         - createpvc-2
         - pvc-to-model-op
+        - run-final-eval-op
         inputs:
           parameters:
             pvc_name:
@@ -1581,6 +1583,7 @@ root:
         dependentTasks:
         - createpvc
         - pvc-to-model-op
+        - run-final-eval-op
         inputs:
           parameters:
             pvc_name: