Address comment

Netflix · Dec 10, 2024 · 746422b · 746422b
1 parent f57bda3
commit 746422b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/metaflow/runner/metaflow_runner.py b/metaflow/runner/metaflow_runner.py
@@ -272,6 +272,9 @@ async def __aenter__(self) -> "Runner":
 
     def __get_executing_run(self, attribute_file_fd, command_obj):
         content = handle_timeout(attribute_file_fd, command_obj, self.file_read_timeout)
+
+        command_obj.sync_wait()
+
         content = json.loads(content)
         pathspec = "%s/%s" % (content.get("flow_name"), content.get("run_id"))
 
@@ -326,7 +329,6 @@ def run(self, **kwargs) -> ExecutingRun:
                 show_output=self.show_output,
             )
             command_obj = self.spm.get(pid)
-            command_obj.sync_wait()
 
             return self.__get_executing_run(attribute_file_fd, command_obj)
 
@@ -358,7 +360,6 @@ def resume(self, **kwargs):
                 show_output=self.show_output,
             )
             command_obj = self.spm.get(pid)
-            command_obj.sync_wait()
 
             return self.__get_executing_run(attribute_file_fd, command_obj)