"""Gentropy package."""

from __future__ import annotations

import warnings

# NOTE: Suppress DeprecationWarnings from pyspark related to pandas API on Spark due to LooseVersion being deprecated in Python 3.12+
warnings.filterwarnings(
    "ignore",
    category=DeprecationWarning,
    module="pyspark.sql.pandas.utils",
)

from gentropy.common.session import Session
from gentropy.dataset.biosample_index import BiosampleIndex
from gentropy.dataset.colocalisation import Colocalisation
from gentropy.dataset.study_index import StudyIndex
from gentropy.dataset.study_locus import StudyLocus
from gentropy.dataset.summary_statistics import SummaryStatistics
from gentropy.dataset.summary_statistics_qc import SummaryStatisticsQC
from gentropy.dataset.target_index import TargetIndex
from gentropy.dataset.variant_index import VariantIndex

__all__ = [
    "Session",
    "StudyIndex",
    "StudyLocus",
    "Colocalisation",
    "BiosampleIndex",
    "SummaryStatistics",
    "SummaryStatisticsQC",
    "VariantIndex",
    "TargetIndex",
]
