TEST-#7421: Fix unidist with APT-installed MPI (#7423)

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

.github/workflows/ci.yml

@@ -115,15 +115,21 @@ jobs:
         run: |
           MODIN_ENGINE=dask python -c "import modin.pandas as pd; print(pd.DataFrame([1,2,3]))"
           MODIN_ENGINE=ray python -c "import modin.pandas as pd; print(pd.DataFrame([1,2,3]))"
-      # - name: Ensure MPI engine start up
-      # Install a working MPI implementation beforehand so mpi4py can link to it.
-      # TODO(https://github.com/modin-project/modin/issues/7421): Find a
-      # way to make MPI installed via APT work. Then uncomment this section.
-      #   run: |
-      #     sudo apt install libmpich-dev
-      #     python -m pip install -e ".[mpi]"
-      #     MODIN_ENGINE=unidist UNIDIST_BACKEND=mpi mpiexec -n 1 python -c "import modin.pandas as pd; print(pd.DataFrame([1,2,3]))"
-      #   if: matrix.os == 'ubuntu'
+      - name: Ensure MPI engine start up
+        # Install a working MPI implementation beforehand so mpi4py can link to it
+        run: |
+          sudo apt-get update
+          sudo apt-get install software-properties-common
+
+          sudo add-apt-repository "deb http://archive.ubuntu.com/ubuntu jammy main universe restricted multiverse"
+          sudo add-apt-repository "deb http://archive.ubuntu.com/ubuntu jammy-updates main universe restricted multiverse"
+          sudo add-apt-repository "deb http://security.ubuntu.com/ubuntu jammy-security main universe restricted multiverse"
+          sudo apt-get update
+
+          sudo apt-get install libmpich-dev=4.0-3 libmpich12=4.0-3 mpich=4.0-3
+          python -m pip install -e ".[mpi]"
+          MODIN_ENGINE=unidist UNIDIST_BACKEND=mpi mpiexec -n 1 python -c "import modin.pandas as pd; print(pd.DataFrame([1,2,3]))"
+        if: matrix.os == 'ubuntu'
 
   test-internals:
     needs: [lint-flake8, python-filter]