crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 21:55:41 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Clean up imports in openmc/deplete/__init__.py

Browse source
This commit is contained in:
Andrew Johnson 2019-06-25 13:26:01 -05:00
parent 6370b362c7
commit 830c0237a8
No known key found for this signature in database
GPG key ID: 253418E91B7F6FEB
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
openmc/deplete/__init__.py
View file

@ -4,6 +4,9 @@ openmc.deplete
A depletion front-end tool.
"""
from sys import exit
from h5py import get_config
from .dummy_comm import DummyCommunicator
@ -12,17 +15,14 @@ try:
comm = MPI.COMM_WORLD
have_mpi = True
# check if running with MPI and if hdf5 is MPI-enabled
from h5py import get_config
# check if running with MPI and if using parallel HDF5
if not get_config().mpi and comm.size > 1:
# Raise exception only on process 0
if comm.rank:
from sys import exit
exit()
raise RuntimeError(
"Need MPI-enabled HDF5 install to perform depletion with MPI"
"Need parallel HDF5 installed to perform depletion with MPI"
)
except ImportError:
comm = DummyCommunicator()