Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
thorstenhater committed Jan 16, 2025
1 parent 307e03a commit 1d0dfd1
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 0 deletions.
2 changes: 2 additions & 0 deletions python/test/fixtures.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,6 +75,7 @@ def repo_path():
def _finalize_mpi():
if _mpi4py_enabled:
from mpi4py import MPI

MPI.Finalize()
else:
A.mpi_finalize()
Expand All @@ -88,6 +89,7 @@ def context():
if _mpi_enabled:
if _mpi4py_enabled:
from mpi4py import MPI

if not MPI.Is_initialized():
MPI.Initialize()
atexit.register(_finalize_mpi)
Expand Down
1 change: 1 addition & 0 deletions python/test/unit/test_diffusion.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,7 @@ def __init__(self, args):
mpi = None
if A.config()["mpi"]:
from mpi4py import MPI

mpi = MPI.COMM_WORLD
gpu_id = None
if A.config()["gpu"]:
Expand Down

0 comments on commit 1d0dfd1

Please sign in to comment.