From a284db614ad4256ae282d17824b5eea180cb5bb7 Mon Sep 17 00:00:00 2001
From: Saurabh Garg <144727284+iamsgarg-ob@users.noreply.github.com>
Date: Wed, 19 Jun 2024 14:21:38 -0700
Subject: [PATCH] [argo] add terminationMessagePolicy (#50) - add a
 terminationMessagePolicy to FallbackToLogsOnError

---
 metaflow/plugins/argo/argo_workflows.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/metaflow/plugins/argo/argo_workflows.py b/metaflow/plugins/argo/argo_workflows.py
index adc922635b2..22e0e425f75 100644
--- a/metaflow/plugins/argo/argo_workflows.py
+++ b/metaflow/plugins/argo/argo_workflows.py
@@ -2015,6 +2015,7 @@ def _container_templates(self):
                         kubernetes_sdk.V1Container(
                             name=self._sanitize(node.name),
                             command=cmds,
+                            termination_message_policy= "FallbackToLogsOnError",
                             ports=[kubernetes_sdk.V1ContainerPort(container_port=port)]
                             if port
                             else None,