Fix missing test in `torch_job` (#33593)

Commit

1 year ago

Fix missing test in `torch_job` (#33593) fix missing tests Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>