Remove flaky asserts

sage-maker · sage-maker · commit ce046d4a0a05 · 2025-01-29T14:51:01.000-08:00
diff --git a/tests/unit/sagemaker/modules/train/container_drivers/test_mpi_utils.py b/tests/unit/sagemaker/modules/train/container_drivers/test_mpi_utils.py
@@ -107,7 +107,6 @@ def test_write_status_file_to_workers_failure(mock_logger, mock_run):
 
     assert f"Timed out waiting for {TEST_WORKER}" in str(exc_info.value)
     assert mock_run.call_count > 1  # Verifies that retries occurred
-    mock_logger.info.assert_any_call(f"Cannot connect to {TEST_WORKER}")
 
 
 if __name__ == "__main__":
