diff --git a/heparchy/read/__init__.py b/heparchy/read/__init__.py index 881e321..26ee346 100644 --- a/heparchy/read/__init__.py +++ b/heparchy/read/__init__.py @@ -9,12 +9,13 @@ ----- Currently only implemented for hepmc and hdf5. """ -import heparchy.read.hdf as hdf +from . import hdf +from .hdf import * import heparchy.read.hepmc as hepmc -HdfReader = hdf.HdfReader HepMC = hepmc.HepMC -__all__ = ["HdfReader", "HepMC"] +__all__ = ["HepMC"] +__all__.extend(hdf.__all__) diff --git a/heparchy/write/__init__.py b/heparchy/write/__init__.py index 396481b..72358eb 100644 --- a/heparchy/write/__init__.py +++ b/heparchy/write/__init__.py @@ -9,9 +9,8 @@ ----- Currently only implemented for hdf5. """ -import heparchy.write.hdf as hdf +from . import hdf +from .hdf import * -HdfWriter = hdf.HdfWriter - -__all__ = ["HdfWriter"] +__all__ = hdf.__all__.copy()