diff --git a/worker/lyra_worker/main.py b/worker/lyra_worker/main.py index 0e5c1d8..165135c 100644 --- a/worker/lyra_worker/main.py +++ b/worker/lyra_worker/main.py @@ -15,7 +15,7 @@ from lyra_worker.library import clear_staging_root from lyra_worker.monitor import MonitorConfig, fulfill_owned_releases, reconcile, sweep from lyra_worker.pipeline import run_pipeline from lyra_worker.registry import ( - build_adapters, build_browser, build_probe, build_resolver, + build_adapters, build_album_popularity, build_browser, build_probe, build_resolver, build_similarity_sources, build_tagger, ) from lyra_worker.scan import build_worklist, clear_worklist, scan_chunk @@ -115,8 +115,10 @@ def maybe_run_scan(conn, resolver, probe, browser, config, dest_root: str = DEST def _run_discovery(conn, sources, browser, config=None) -> DiscoveryResult: - cfg = DiscoveryConfig.from_config(config if config is not None else get_config(conn)) - return run_discovery(conn, sources, browser, cfg) + resolved = config if config is not None else get_config(conn) + cfg = DiscoveryConfig.from_config(resolved) + popularity = build_album_popularity(resolved) + return run_discovery(conn, sources, browser, cfg, popularity=popularity) def finish_job(conn, job_id, mcfg) -> None: