"""
QKMeans: Fast k-means++ seeding under the manifold hypothesis.

This package implements the QKMeans algorithm from:
"Faster k-means Seeding Under The Manifold Hypothesis" (ICML 2026)

The algorithm provides accelerated k-means++ seeding using rejection sampling,
with theoretical guarantees under the manifold hypothesis.
"""

from ._qkmeans import QKMeansSeeding

# Import C++ backend utilities
try:
    from ._qkmeans_cpp import (
        ANNSType,
        QKMeansConfig,
        QKMeansResult,
        get_simd_level,
        is_anns_available,
    )
except ImportError as e:
    raise ImportError(
        "Failed to import QKMeans C++ backend. "
        "Please ensure the package is properly installed."
    ) from e

__version__ = "0.1.0"

__all__ = [
    "QKMeansSeeding",
    "ANNSType",
    "QKMeansConfig",
    "QKMeansResult",
    "get_simd_level",
    "is_anns_available",
]
