Fix issue with setting metadata when using the runner (#1875)

Netflix · Jun 4, 2024 · 97d3f58 · 97d3f58
1 parent 9d3516c
commit 97d3f58
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/metaflow/cli.py b/metaflow/cli.py
@@ -10,7 +10,6 @@
 
 from . import decorators, lint, metaflow_version, namespace, parameters, plugins
 from .cli_args import cli_args
-from .client.core import get_metadata
 from .datastore import FlowDataStore, TaskDataStore, TaskDataStoreSet
 from .exception import CommandException, MetaflowException
 from .graph import FlowGraph
@@ -700,7 +699,12 @@ def resume(
     runtime.persist_constants()
     write_file(
         runner_attribute_file,
-        "%s:%s" % (get_metadata(), "/".join((obj.flow.name, runtime.run_id))),
+        "%s@%s:%s"
+        % (
+            obj.metadata.__class__.TYPE,
+            obj.metadata.__class__.INFO,
+            "/".join((obj.flow.name, runtime.run_id)),
+        ),
     )
     if clone_only:
         runtime.clone_original_run()
@@ -763,7 +767,12 @@ def run(
     runtime.persist_constants()
     write_file(
         runner_attribute_file,
-        "%s:%s" % (get_metadata(), "/".join((obj.flow.name, runtime.run_id))),
+        "%s@%s:%s"
+        % (
+            obj.metadata.__class__.TYPE,
+            obj.metadata.__class__.INFO,
+            "/".join((obj.flow.name, runtime.run_id)),
+        ),
     )
     runtime.execute()