From aef35320e9ad43e2abb518a52ecc430125ce561b Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Fri, 15 Nov 2024 16:04:21 -0500
Subject: [PATCH] [r2] CI: fix Python workflow (#4364)

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **Chores**
- Updated the workflow configuration for testing Python dependencies to
improve clarity and installation steps.
- Adjusted installation commands for `mpi4py` and `horovod` to
streamline the process.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
---
 .github/workflows/test_python.yml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/test_python.yml b/.github/workflows/test_python.yml
index 956c91e491..089b191c23 100644
--- a/.github/workflows/test_python.yml
+++ b/.github/workflows/test_python.yml
@@ -34,7 +34,9 @@ jobs:
         # changes, setting `TENSORFLOW_ROOT`.
         TENSORFLOW_VERSION: ${{ matrix.tf }}
         DP_BUILD_TESTING: 1
-    - run: source/install/uv_with_retry.sh pip install --system --no-build-isolation horovod mpi4py
+    - run: |
+        source/install/uv_with_retry.sh pip install --system mpi4py
+        source/install/uv_with_retry.sh pip install --system --no-build-isolation horovod
       if: ${{ matrix.tf == '' }}
       env:
         HOROVOD_WITH_TENSORFLOW: 1