Skip to content

fix: check all speaker diarization imports to prevent silent failures #2824

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
AniruddhaKhandare wants to merge 2 commits into from
Closed

fix: check all speaker diarization imports to prevent silent failures #2824

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b84d32b
fix: check all speaker diarization imports to prevent silent failures
AniruddhaKhandare Feb 28, 2026
fb61a84
fix: remove redundant logging import in except block
AniruddhaKhandare Feb 28, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 14 additions & 2 deletions funasr/auto/auto_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -46,11 +46,18 @@ def _resolve_ncpu(config, fallback=4):
value = fallback
return max(value, 1)

sv_chunk = None
postprocess = None
distribute_spk = None
ClusterBackend = None
try:
from funasr.models.campplus.utils import sv_chunk, postprocess, distribute_spk
from funasr.models.campplus.cluster_backend import ClusterBackend
except:
pass
except Exception as e:
logging.warning(
f"Speaker diarization modules could not be imported: {e}. "
"Speaker diarization functionality will be unavailable."
)


def prepare_data_iterator(data_in, input_len=None, data_type=None, key=None):
Expand Down Expand Up @@ -175,6 +182,11 @@ def __init__(self, **kwargs):
spk_kwargs["device"] = kwargs["device"]
spk_kwargs.setdefault("ncpu", kwargs.get("ncpu", 4))
spk_model, spk_kwargs = self.build_model(**spk_kwargs)
if any(x is None for x in (sv_chunk, postprocess, distribute_spk, ClusterBackend)):
raise ImportError(
"Speaker diarization dependencies failed to import. "
"Please ensure scikit-learn>=1.3.0 and scipy are properly installed."
)
self.cb_model = ClusterBackend(**cb_kwargs).to(kwargs["device"])
spk_mode = kwargs.get("spk_mode", "punc_segment")
if spk_mode not in ["default", "vad_segment", "punc_segment"]:
Expand Down