Add params to connect to zipkin tracing platform

Netflix · Nov 13, 2023 · 847b377 · 847b377
1 parent 5ece011
commit 847b377
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/metaflow/runtime.py b/metaflow/runtime.py
@@ -219,6 +219,13 @@ def execute(self):
                 "Workflow starting (run-id %s):" % self._run_id, system_msg=True
             )
 
+        if TRACING_URL_TEMPLATE and tracing.get_trace_id():
+            tracing_url = TRACING_URL_TEMPLATE.format(trace_id=tracing.get_trace_id())
+            self._logger(
+                "Tracing URL will appear in the following %s. " % tracing_url,
+                system_msg=True,
+            )               
+
         self._metadata.start_run_heartbeat(self._flow.name, self._run_id)
 
         if self._params_task:
@@ -309,13 +316,7 @@ def execute(self):
                     system_msg=True,
                 )
             else:
-                self._logger("Done!", system_msg=True)
-            if TRACING_URL_TEMPLATE and tracing.get_trace_id():
-                tracing_log_template = "Tracing URL is %s" % TRACING_URL_TEMPLATE
-                self._logger(
-                    tracing_log_template.format(trace_id=tracing.get_trace_id()),
-                    system_msg=True,
-                )                
+                self._logger("Done!", system_msg=True)   
         elif self._clone_only:
             self._logger(
                 "Clone-only resume complete -- only previously successful steps were "