restore missing import, add __all__

Jhsmit · Jun 14, 2024 · ca50606 · ca50606
1 parent e8fb69a
commit ca50606
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/pyhdx/__init__.py b/pyhdx/__init__.py
@@ -1,8 +1,28 @@
 from pyhdx.__version__ import __version__
+from pyhdx.datasets import read_dynamx
+from pyhdx.fitting_torch import TorchFitResult, TorchFitResultSet
+from pyhdx.models import (
+    Coverage,
+    HDXMeasurement,
+    HDXMeasurementSet,
+    HDXTimepoint,
+)
 
 VERSION_STRING = f"PyHDX {__version__}"
 
 try:
     from pyhdx.output import FitReport
 except ModuleNotFoundError:
     pass
+
+
+__all__ = [
+    "HDXTimepoint",
+    "HDXMeasurement",
+    "Coverage",
+    "HDXMeasurementSet",
+    "read_dynamx",
+    "TorchFitResult",
+    "TorchFitResultSet",
+    "FitReport",
+]