Remove flaky asserts

aws · Jan 29, 2025 · ce046d4 · ce046d4
1 parent 4fd9f8c
commit ce046d4
Showing 1 changed file with 0 additions and 1 deletion.
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__":