[AI Codemod][DevmateFBSourceTestFailureBot] Fix for T241916639 ("Your diff, D84932408, broke one test") (#166168)

Reviewed By: XilunWu

Differential Revision: D84983164

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166168
Approved by: https://github.com/H-Huang, https://github.com/fduwjj
This commit is contained in:
Xilun Wu 2025-10-25 06:46:23 +00:00 committed by PyTorch MergeBot
parent c9bc00f016
commit 661a56002f

View File

@ -257,7 +257,10 @@ class RecordWorkerEventsTest(unittest.TestCase):
self.assertEqual(failed_event.metadata["state"], "FAILED") self.assertEqual(failed_event.metadata["state"], "FAILED")
self.assertEqual(failed_event.metadata["global_rank"], 0) self.assertEqual(failed_event.metadata["global_rank"], 0)
md = json.loads(failed_event.metadata["metadata"]) md = json.loads(failed_event.metadata["metadata"])
self.assertEqual(failed_event.metadata["raw_error"], '{"message": "<NONE>"}') self.assertEqual(
failed_event.metadata["raw_error"],
'{"message": "<NONE>", "errorTraits": {"category": "system_terminated_error", "retryability": "False"}}',
)
self.assertEqual(md["exit_code"], [1]) self.assertEqual(md["exit_code"], [1])
self.assertEqual(md["worker_pid"], [1000]) self.assertEqual(md["worker_pid"], [1000])