PDGrapher/src/pdgrapher/__init__.py

from .datasets import Dataset
from .pdgrapher import PDGrapher
from .train import Trainer

__version__ = "0.1.0"

__all__ = ["PDGrapher", "Dataset", "Trainer", "__version__"]