address nitpick

Netflix · Jan 16, 2025 · c402e03 · c402e03
1 parent ed0b639
commit c402e03
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/metaflow/plugins/argo/argo_workflows.py b/metaflow/plugins/argo/argo_workflows.py
@@ -1705,7 +1705,7 @@ def _container_templates(self):
                     },
                     **{
                         # Some optional values for bookkeeping
-                        "MF_FLOW_FILENAME": os.path.basename(sys.argv[0]),
+                        "METAFLOW_FLOW_FILENAME": os.path.basename(sys.argv[0]),
                         "METAFLOW_FLOW_NAME": self.flow.name,
                         "METAFLOW_STEP_NAME": node.name,
                         "METAFLOW_RUN_ID": run_id,

diff --git a/metaflow/plugins/kubernetes/kubernetes_cli.py b/metaflow/plugins/kubernetes/kubernetes_cli.py
@@ -190,7 +190,7 @@ def echo(msg, stream="stderr", job_id=None, **kwargs):
     executable = ctx.obj.environment.executable(step_name, executable)
 
     # Set environment
-    env = {"MF_FLOW_FILENAME": os.path.basename(sys.argv[0])}
+    env = {"METAFLOW_FLOW_FILENAME": os.path.basename(sys.argv[0])}
     env_deco = [deco for deco in node.decorators if deco.name == "environment"]
     if env_deco:
         env = env_deco[0].attributes["vars"]

diff --git a/metaflow/plugins/kubernetes/spot_monitor_sidecar.py b/metaflow/plugins/kubernetes/spot_monitor_sidecar.py
@@ -78,7 +78,7 @@ def _monitor_loop(self):
             time.sleep(self.POLL_INTERVAL)
 
     def _emit_termination_metadata(self, termination_time):
-        flow_filename = os.getenv("MF_FLOW_FILENAME")
+        flow_filename = os.getenv("METAFLOW_FLOW_FILENAME")
         pathspec = os.getenv("MF_PATHSPEC")
         _, run_id, step_name, task_id = pathspec.split("/")
         retry_count = os.getenv("MF_ATTEMPT")