"""Unit test: load SegmentationModel weights from HF mlx-community repo."""
import pytest
from pyannote_diarization_3_1_mlx.segmentation import SegmentationModel


@pytest.mark.integration
def test_segmentation_loads_from_hf():
    m = SegmentationModel.from_hf()
    assert m is not None