Try fix

Author: amrit110

.github/workflows/integration_tests.yml

@@ -38,25 +38,23 @@ jobs:
   integration-tests:
     runs-on: [self-hosted, gpu, db]
     steps:
-      - name: Export OpenMPI PATH
-        run: |
-          export PATH=/opt/openmpi-4.1.5/bin:$PATH
-          export LD_LIBRARY_PATH=/opt/openmpi-4.1.5/lib:$LD_LIBRARY_PATH
-          ompi_info
-      - uses: actions/checkout@v3
       - name: Install poetry
+      - uses: actions/checkout@v3
         run: pip install poetry
       - uses: actions/setup-python@v4.7.1
         with:
           python-version: '3.10'
       - name: Install dependencies and check code
         run: |
+          export PATH=/opt/openmpi-4.1.5/bin:$PATH
+          export LD_LIBRARY_PATH=/opt/openmpi-4.1.5/lib:$LD_LIBRARY_PATH
           ompi_info
           poetry env use '3.10'
           source $(poetry env info --path)/bin/activate
-          env MPICC=/opt/openmpi-4.1.5/bin/mpicc python3 -m pip install git+https://github.com/mpi4py/mpi4py
           poetry install --with test,dev --all-extras
           coverage run -m pytest -m integration_test && coverage xml && coverage report -m
+      - env:
+          MPICC: /opt/openmpi-4.1.5/bin/mpicc
       - name: Upload coverage to Codecov
         uses: Wandalen/wretry.action@v1.0.36
         with: