| # Copyright (c) 2026 The Anuvani-Diarization authors | |
| # SPDX-License-Identifier: MIT OR Apache-2.0 | |
| """The frozen release package (release/anuvani_diar_v1/anuvani_diar_ref), imported from its own folder under the alias | |
| `anuvani_diar_ref` so it can sit beside the new `anuvani_diar` package. Nothing is copied or edited.""" | |
| import importlib.util | |
| import os | |
| import sys | |
| from pathlib import Path | |
| # ANUVANI_DIAR_MAIN_TREE: see conftest.py (a linked git worktree reads the release's weights in the main checkout) | |
| REL = Path(os.environ.get("ANUVANI_DIAR_MAIN_TREE") or Path(__file__).resolve().parents[3]) / "release/anuvani_diar_v1" | |
| def load(): | |
| if "anuvani_diar_ref" in sys.modules: | |
| return sys.modules["anuvani_diar_ref"] | |
| spec = importlib.util.spec_from_file_location("anuvani_diar_ref", REL / "anuvani_diar_ref/__init__.py", | |
| submodule_search_locations=[str(REL / "anuvani_diar_ref")]) | |
| mod = importlib.util.module_from_spec(spec) | |
| sys.modules["anuvani_diar_ref"] = mod | |
| spec.loader.exec_module(mod) | |
| from anuvani_diar_ref import vbx | |
| vbx._HUB[0] = REL / "bundle/hub" | |
| return mod | |