Lint

Cambridge-ICCS · Jan 13, 2025 · 8c27dc1 · 8c27dc1
1 parent 5fa7801
commit 8c27dc1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/3_MultiGPU/multigpu_infer_python.py b/examples/3_MultiGPU/multigpu_infer_python.py
@@ -3,9 +3,11 @@
 import torch
 try:
     from mpi4py import MPI
+
     rank = MPI.COMM_WORLD.rank
 except ModuleNotFoundError:
     from warnings import warn
+
     warn("Running with rank 0 under the assumption that MPI is not being used.")
     rank = 0