"""
PPMI - Pointwise Mutual Information Library
A Python library for calculating Pointwise Mutual Information (PMI) and Positive PMI.
"""

from .pmi import (
    PMI,
    calculate_pmi_from_probabilities,
    calculate_ppmi_from_probabilities,
    calculate_pmi_from_counts,
    calculate_ppmi_from_counts
)

__version__ = "0.1.1"
__all__ = [
    "PMI",
    "calculate_pmi_from_probabilities",
    "calculate_ppmi_from_probabilities",
    "calculate_pmi_from_counts",
    "calculate_ppmi_from_counts"
]
