diff --git a/src/utils/openmc/opencg_compatible.py b/src/utils/openmc/opencg_compatible.py index 320529f68a..c0e6d595d0 100644 --- a/src/utils/openmc/opencg_compatible.py +++ b/src/utils/openmc/opencg_compatible.py @@ -1,7 +1,12 @@ import copy import numpy as np -import opencg + +try: + import opencg +except ImportError: + msg = 'Unable to import opencg which is needed by openmc.opencg_compatible' + raise ImportError(msg) import openmc diff --git a/src/utils/openmc/summary.py b/src/utils/openmc/summary.py index da10690089..fec616c064 100644 --- a/src/utils/openmc/summary.py +++ b/src/utils/openmc/summary.py @@ -1,12 +1,11 @@ import numpy as np import openmc -from openmc.opencg_compatible import get_opencg_geometry try: import h5py except ImportError: - msg = 'Unable to import h5py which is necessary for openmc.summary' + msg = 'Unable to import h5py which is needed by openmc.summary' raise ImportError(msg) @@ -552,6 +551,14 @@ class Summary(object): def make_opencg_geometry(self): + + try: + from openmc.opencg_compatible import get_opencg_geometry + except ImportError: + msg = 'Unable to import opencg which is needed ' \ + 'by Summary.make_opencg_geometry()' + raise ImportError(msg) + if self.opencg_geometry is None: self.opencg_geometry = get_opencg_geometry(self.openmc_geometry)