model/heads/__init__.py

"""
`model.heads` package exports.

This file exposes the commonly used head classes so callers can
import them from `model.heads` directly, e.g.:

	from model.heads import NERHead, QAHead, SentimentHead

Keeping an explicit `__all__` makes the package's public API clear.
"""

from .ner_head import NERHead
from .qa_head import QAHead
from .sentiment_head import SentimentHead

__all__ = ["NERHead", "QAHead", "SentimentHead"]