Catch issue with partial mpi4py installation. #21353

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

deependujha merged 7 commits into Lightning-AI:master from roccomoretti:mpi_checking

Nov 19, 2025

src/lightning/fabric/plugins/environments/mpi.py

-Original file line number
+Diff line change
@@ Expand Up / @@ -73,7 +73,10 @@ def detect() -> bool: @@
             if not _MPI4PY_AVAILABLE:
                 return False
-            from mpi4py import MPI
+            try:
+                from mpi4py import MPI
+            except ImportError:
+                return False
             return MPI.COMM_WORLD.Get_size() > 1
@@ Expand Down @@