diff --git a/openmc/deplete/__init__.py b/openmc/deplete/__init__.py index 8c19615ece..f0b04c3de5 100644 --- a/openmc/deplete/__init__.py +++ b/openmc/deplete/__init__.py @@ -35,7 +35,10 @@ from .nuclide import * from .chain import * from .operator import * from .reaction_rates import * -from .abc import * +from .atom_number import * from .results import * from .results_list import * from .integrators import * +from . import abc +from . import cram +from . import helpers diff --git a/openmc/deplete/abc.py b/openmc/deplete/abc.py index e0ed64645e..565b9e2dcf 100644 --- a/openmc/deplete/abc.py +++ b/openmc/deplete/abc.py @@ -24,6 +24,13 @@ from .results import Results from .chain import Chain from .results_list import ResultsList + +__all__ = [ + "OperatorResult", "TransportOperator", "ReactionRateHelper", + "EnergyHelper", "FissionYieldHelper", "TalliedFissionYieldHelper", + "Integrator", "SIIntegrator"] + + OperatorResult = namedtuple('OperatorResult', ['k', 'rates']) OperatorResult.__doc__ = """\ Result of applying transport operator diff --git a/openmc/deplete/chain.py b/openmc/deplete/chain.py index 510112c7c9..a2f7db8184 100644 --- a/openmc/deplete/chain.py +++ b/openmc/deplete/chain.py @@ -45,6 +45,8 @@ _REACTIONS = [ ] +__all__ = ["Chain"] + def replace_missing(product, decay_data): """Replace missing product with suitable decay daughter. diff --git a/openmc/deplete/cram.py b/openmc/deplete/cram.py index 67a455cb35..31049bb724 100644 --- a/openmc/deplete/cram.py +++ b/openmc/deplete/cram.py @@ -13,8 +13,6 @@ import scipy.sparse.linalg as sla from . import comm -__all__ = ["deplete", "timed_deplete", "CRAM16", "CRAM48"] - def deplete(chain, x, rates, dt, matrix_func=None): """Deplete materials using given reaction rates for a specified time diff --git a/openmc/deplete/nuclide.py b/openmc/deplete/nuclide.py index 8896180a88..ec1668dacb 100644 --- a/openmc/deplete/nuclide.py +++ b/openmc/deplete/nuclide.py @@ -17,6 +17,10 @@ from numpy import empty from openmc.checkvalue import check_type +__all__ = [ + "DecayTuple", "ReactionTuple", "Nuclide", "FissionYield", + "FissionYieldDistribution"] + DecayTuple = namedtuple('DecayTuple', 'type target branching_ratio') DecayTuple.__doc__ = """\ diff --git a/openmc/deplete/operator.py b/openmc/deplete/operator.py index 86dbeb7550..9800bb2805 100644 --- a/openmc/deplete/operator.py +++ b/openmc/deplete/operator.py @@ -30,6 +30,9 @@ from .helpers import ( FissionYieldCutoffHelper, AveragedFissionYieldHelper) +__all__ = ["Operator", "OperatorResult"] + + def _distribute(items): """Distribute items across MPI communicator diff --git a/openmc/deplete/reaction_rates.py b/openmc/deplete/reaction_rates.py index cea2f19976..85c8d8998d 100644 --- a/openmc/deplete/reaction_rates.py +++ b/openmc/deplete/reaction_rates.py @@ -7,6 +7,9 @@ from collections import OrderedDict import numpy as np +__all__ = ["ReactionRates"] + + class ReactionRates(np.ndarray): """Reaction rates resulting from a transport operator call diff --git a/openmc/deplete/results.py b/openmc/deplete/results.py index 6593d5fc5f..b44739bc5f 100644 --- a/openmc/deplete/results.py +++ b/openmc/deplete/results.py @@ -16,6 +16,9 @@ from .reaction_rates import ReactionRates _VERSION_RESULTS = (1, 0) +__all__ = ["Results"] + + class Results(object): """Output of a depletion run diff --git a/openmc/deplete/results_list.py b/openmc/deplete/results_list.py index 3b5166ed8a..26bc8785fe 100644 --- a/openmc/deplete/results_list.py +++ b/openmc/deplete/results_list.py @@ -5,6 +5,9 @@ from .results import Results, _VERSION_RESULTS from openmc.checkvalue import check_filetype_version +__all__ = ["ResultsList"] + + class ResultsList(list): """A list of openmc.deplete.Results objects