address changes

Netflix · Apr 24, 2023 · 89b13e8 · 89b13e8
1 parent 3694a19
commit 89b13e8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/metaflow/plugins/argo/argo_workflows.py b/metaflow/plugins/argo/argo_workflows.py
@@ -969,6 +969,13 @@ def _container_templates(self):
                         "METAFLOW_RUNTIME_ENVIRONMENT": "kubernetes",
                         "METAFLOW_OWNER": self.username,
                     },
+                    **{
+                        # Configuration for Argo Events
+                        "METAFLOW_ARGO_EVENTS_EVENT": ARGO_EVENTS_EVENT,
+                        "METAFLOW_ARGO_EVENTS_EVENT_BUS": ARGO_EVENTS_EVENT_BUS,
+                        "METAFLOW_ARGO_EVENTS_EVENT_SOURCE": ARGO_EVENTS_EVENT_SOURCE,
+                        "METAFLOW_ARGO_EVENTS_SERVICE_ACCOUNT": ARGO_EVENTS_SERVICE_ACCOUNT,
+                    },
                     **{
                         # Some optional values for bookkeeping
                         "METAFLOW_FLOW_NAME": self.flow.name,