From 06b93133e137df06294dbbbcd4d7e911c9ca53f1 Mon Sep 17 00:00:00 2001 From: zdevito Date: Mon, 22 Dec 2025 14:39:18 -0800 Subject: [PATCH] Fix oss build Differential Revision: [D89695534](https://our.internmc.facebook.com/intern/diff/D89695534/) **NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D89695534/)! [ghstack-poisoned] --- python/tests/test_job.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/python/tests/test_job.py b/python/tests/test_job.py index b9533346c..416bc03fe 100644 --- a/python/tests/test_job.py +++ b/python/tests/test_job.py @@ -395,11 +395,10 @@ def test_train_script_job_state_batch(): try: job = LocalJob(("batch_launched_hosts",)) job.apply(client_script=train_script) + status = job.process.wait() stdout = open(os.path.join(job._log_dir, "stdout.log"), "r").read() stderr = open(os.path.join(job._log_dir, "stderr.log"), "r").read() - assert ( - 0 == job.process.wait() - ), f"Job failed\nstdout:\n{stdout}\nstderr:\n{stderr}" + assert status == 0, f"Job failed\nstdout:\n{stdout}\nstderr:\n{stderr}" assert "batch_launched_hosts True" in stdout # look in job._log_dir for the stdout file which will have the batch_lauched_hosts True finally: