From ca506060342a65927314acb3d33ee2e950b07ad5 Mon Sep 17 00:00:00 2001 From: Jochem Smit Date: Fri, 14 Jun 2024 10:23:52 +0200 Subject: [PATCH] restore missing import, add __all__ --- pyhdx/__init__.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/pyhdx/__init__.py b/pyhdx/__init__.py index 3a67c7a0..acdffeb2 100644 --- a/pyhdx/__init__.py +++ b/pyhdx/__init__.py @@ -1,4 +1,12 @@ 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__}" @@ -6,3 +14,15 @@ from pyhdx.output import FitReport except ModuleNotFoundError: pass + + +__all__ = [ + "HDXTimepoint", + "HDXMeasurement", + "Coverage", + "HDXMeasurementSet", + "read_dynamx", + "TorchFitResult", + "TorchFitResultSet", + "FitReport", +]