Wi$ 8ddlZddedejfdZdZy)Nfunc embeddingscJ ddl}|dk(rt ||j r|j ||\}}||fStt|j}d|vr#d|vrddl m}|j ||\}}||fS|j|}|dfS|d k(rit ||j r|j|Stt|j}d|vrd|vrddl m}|j|Sy|d k(rot ||j r|j||}|Stt|j}d|vrd|vrddl m}|j||}|Syy#ttf$rtddddi}YwxYw) aFunction used to select the HDBSCAN-like model for generating predictions and probabilities. Arguments: model: The cluster model. func: The function to use. Options: - "approximate_predict" - "all_points_membership_vectors" - "membership_vector" embeddings: Input embeddings for "approximate_predict" and "membership_vector" rNhdbscanHDBSCANapproximate_predictcuml)rall_points_membership_vectorsmembership_vector) r ImportErrorModuleNotFoundErrortype isinstancerr strlower cuml.clusterpredictr r )modelrrr predictions probabilitiesstr_type_model cuml_hdbscans c/home/sietch6/trending-topics-pipeline/venv/lib/python3.12/site-packages/bertopic/cluster/_utils.pyhdbscan_delegatorrs;  $$ eW__ -)0)D)DUJ)W &K - -T%[)//1 ^ # ^(C <)5)I)I%Q[)\ &K - -mmJ/ D   .. eW__ -88? ?T%[)//1 ^ # ^(C <==eD D "" eW__ -#55eZHM T%[)//1 ^ # ^(C <(::5*MM #C , -;8$y"y$&78:;sE;;#F"!F"c ddl}t ||j ryt t|j}d|vrd|vryy#ttf$rtddddi}YhwxYw) z@Check whether the input model is a supported HDBSCAN-like model.rNrrrTr F)rr rrrrrr)rrrs ris_supported_hdbscanrCsx;%)e%++-N I$?  , -;8$y"y$&78:;sA #A/.A/)N)numpynprndarrayrrrrr"s%<3<BJJ<~r!