Fix bugs in --until implementation

Fixed several critical bugs in the resume --until feature:

1. Initialize until_steps to None to prevent NameError when --until is not specified
2. Fix error message to show specific invalid step name instead of entire tuple
3. Remove incorrect clone_only error check that had wrong error message
4. Remove debug print statements
5. Improve comment clarity for step precedence logic

Claude helped!

Author: romain-intel

metaflow/cli_components/run_cmds.py

@@ -232,15 +232,13 @@ def resume(
     if step_to_rerun is None:
         steps_to_rerun = set()
     else:
-        if clone_only:
-            raise CommandException("Cannot specify both --clone-only and --until")
         # validate step name
         for s in step_to_rerun:
             if s not in obj.graph.nodes:
                 raise CommandException(
                     "Invalid step name {0} specified, must be step present in "
                     "current form of execution graph. Valid step names include: {1}".format(
-                        step_to_rerun, ",".join(list(obj.graph.nodes.keys()))
+                        s, ",".join(list(obj.graph.nodes.keys()))
                     )
                 )
 
@@ -261,6 +259,7 @@ def resume(
     if clone_only and until is not None:
         raise CommandException("Cannot specify both --clone-only and --until")
 
+    until_steps = None
     if until is not None:
         until_steps = set(until.split(","))
         for step in until_steps:

metaflow/runtime.py

@@ -163,14 +163,10 @@ def __init__(
                     self._steps_no_run.add(next_step)
             elif step_name in self._steps_to_rerun:
                 for next_step in out_funcs:
-                    # We may add things that are in steps_no_run but
-                    # we will remove them later.
                     self._steps_to_rerun.add(next_step)
+        # Remove any steps that should not be run (--until takes precedence)
         self._steps_to_rerun = self._steps_to_rerun - self._steps_no_run
         self._steps_can_clone = all_steps - self._steps_to_rerun - self._steps_no_run
-        print(f"steps_to_rerun: {self._steps_to_rerun}")
-        print(f"steps_no_run: {self._steps_no_run}")
-        print(f"steps_can_clone: {self._steps_can_clone}")
 
         self._origin_ds_set = None
         if clone_run_id: