src/__init__.py

from .model import Tox21SNN
from .features import EnhancedFeatureExtractor, TOX21_TARGETS
from .ensemble import Tox21Ensemble

TASKS = TOX21_TARGETS

__all__ = ["Tox21SNN", "EnhancedFeatureExtractor", "Tox21Ensemble", "TASKS"]