Juan Pablo ZuluagaandJuanPZuluaga
b64ab05d09
Signed-off-by: JuanPZuluaga <juanz9312@gmal.com> Co-authored-by: JuanPZuluaga <juanz9312@gmal.com>
22 lines
495 B
Python
22 lines
495 B
Python
"""Fixtures for the process-wide speaker cache singleton."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def fresh_speaker_cache():
|
|
"""Reset the process-wide speaker cache singleton before and after the test."""
|
|
import vllm_omni.utils.speaker_cache as _sc
|
|
|
|
def _reset():
|
|
with _sc._SINGLETON_LOCK:
|
|
if _sc._SINGLETON is not None:
|
|
_sc._SINGLETON.clear()
|
|
_sc._SINGLETON = None
|
|
|
|
_reset()
|
|
yield
|
|
_reset()
|