__init__.py

from .class1_affinity_predictor import Class1AffinityPredictor
from .class1_neural_network import Class1NeuralNetwork
from .version import __version__

__all__ = [
    "__version__",
    "Class1AffinityPredictor",
    "Class1NeuralNetwork",
]