export all submodule symbols

jacanchaplais · Nov 2, 2022 · dd79cea · dd79cea
1 parent 936b36a
commit dd79cea
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git 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
@@ -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()