Skip to content

feat(scripts): add exec_cmd to ColabRunner and simplify remote execution - #1334

Merged
tushuhei merged 1 commit into
mainfrom
colab-exec-cmd
Aug 17, 2026
Merged

feat(scripts): add exec_cmd to ColabRunner and simplify remote execution#1334
tushuhei merged 1 commit into
mainfrom
colab-exec-cmd

Conversation

@tushuhei

Copy link
Copy Markdown
Member
  • Add exec_code and exec_cmd helper methods to ColabRunner in scripts/colab_runner.py, streaming subprocess stdout via sys.stdout for real-time console metrics.
  • Add pre-provision session cleanup (stop_session) in provision_session to clean up orphan Colab VMs and prevent TooManyAssignmentsError.
  • Simplify Colab execution in scripts/run_training_pipeline.py by calling runner.exec_cmd(remote_args).
  • Add unit test test_exec_cmd in scripts/tests/test_colab_runner.py.

TAG=agy
CONV=da1ee413-f759-43cd-8f2d-90cf430c1f51

- Add exec_code and exec_cmd helper methods to ColabRunner in scripts/colab_runner.py, streaming subprocess stdout via sys.stdout for real-time console metrics.
- Add pre-provision session cleanup (stop_session) in provision_session to clean up orphan Colab VMs and prevent TooManyAssignmentsError.
- Simplify Colab execution in scripts/run_training_pipeline.py by calling runner.exec_cmd(remote_args).
- Add unit test test_exec_cmd in scripts/tests/test_colab_runner.py.

TAG=agy
CONV=da1ee413-f759-43cd-8f2d-90cf430c1f51

@kojiishi kojiishi left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm

@tushuhei
tushuhei merged commit 64b21b5 into main Aug 17, 2026
43 checks passed
@tushuhei
tushuhei deleted the colab-exec-cmd branch August 17, 2026 09:50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants