,jE lSSKJr SSKrSSKJr SSKJrJrJrJ r J r J r J r J r SSKrSSKJrJr SSKJrJr SSKJr SSKr"S S \5r"S S 5rS \\4SjrS \\4SjrS\\ \ \ 4\ 4S\\ \ \ 4\ 4S\S\S\4 Sjr\"5r\"5Hr\RA\5 M g))EnumN)Path)DictListSetOptionalTextTupleTypeUnion) PreprocessorsProtocol)create_protocolget_init)Databasec \rSrSrSrSrSrSrg) LoadingMode,rrN)__name__ __module__ __qualname____firstlineno__OVERRIDEKEEPERROR__static_attributes__rڨC:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\pyannote/database/registry.pyrr,sH D Er rc\rSrSrSrSSjr\R4S\\ \ 4S\4Sjjr \R\ "54S \\ \ 4S\S \\ 4S jjrS\4S jrSS \\S\4SjjrSrS\R4S\S \\ \ 4S\4SjjrSrSrg)RegistrygzDatabase and experimental protocols registry Usage ----- >>> from pyannote.database import registry >>> registry.load_database("/path/to/first/database.yml") >>> registry.load_database("/path/to/second/database.yml") returnNc^[5Ul[5Ul[5UlgN)dictconfigssources databases)selfs r!__init__Registry.__init__qs$B*. 04v ,06r pathmodecTURX[5S9 UR5 g)aqLoad YAML configuration file into the registry Parameters ---------- path : str or Path Path to YAML configuration file. mode : LoadingMode, optional Controls how to handle conflicts in protocol names. Defaults to overriding the existing protocol. Usage ----- >>> from pyannote.database import registry >>> registry.load_database("/path/to/database.yml") r0loadingN)_load_database_helperset_reload_meta_protocols)r,r/r0s r! load_databaseRegistry.load_databases%* ""4CE"B ##%r database_ymlr3cX[U5R5R5nX;agURU5 [ US5n[ R "U[ RS9nSSS5 WRS[55n[U[5(dU/nUHBn[U5nUR5(dURU- nURXrUS9 MD URS[55nURSS5n U bXS'UR!5HupUR#XXS9 M URS [55n U R!5Hup[U [5(dU /n [5nU HLn[U5nUR5(dURU- nUR%['U55 MN XR(['U 5'M XPR*U'g!,(df  GN=f) aHelper function for recursive loading Parameters ---------- database_yml : Union[Text, Path] Path to the database.yml mode : LoadingMode, optional Controls how to handle conflicts in protocol names. Defaults to overriding the existing protocol. Nr)Loader Requirementsr2 ProtocolsXr0 Databases)r expanduserresolveaddopenyamlload SafeLoaderpoplist isinstance is_absoluteparentr4getr(items_load_protocolsappendstrr*r))r,r9r0r3fconfig requirementsrequirement_yaml protocolsxdb_name db_entriesr+value path_listpr/s r!r4Registry._load_database_helpers$L)446>>@   "   L!, $YYq9F%zz.$&9 ,--(>L , #$45 #//11#/#6#69I#I  & &'7G & T !-JJ{DF3  MM#t $ =cN$-??#4 G  l N$5JJ{DF3 'oo/NGeT**#'6IAw''))'..5D  T+  *3LLW &0&, \"U% $s $H H)c URUnU"S0UD6$![a& US:XaSnOSnURUS9n[U5ef=f)zGet database by name Parameters ---------- database_name : str Database name. Returns ------- database : Database Database instance r?zCould not find any meta-protocol. Please refer to pyannote.database documentation to learn how to define them: https://github.com/pyannote/pyannote-databasezCould not find any protocol for "{name}" database. Please refer to pyannote.database documentation to learn how to define them: https://github.com/pyannote/pyannote-database)namer)r+KeyErrorformat ValueError)r, database_namekwargsdatabasemsgs r! get_databaseRegistry.get_database sp "~~m4H$!&!!! "#DQ jjmj4S/ ! "s 0A  preprocessorsc|URS5up4nURU5nURXEUS9nXlU$)aGet protocol by full name Parameters ---------- name : str Protocol full name (e.g. "Etape.SpeakerDiarization.TV") preprocessors : dict or (key, preprocessor) iterable When provided, each protocol item (dictionary) are preprocessed, such that item[key] = preprocessor(item). In case 'preprocessor' is not callable, it should be a string containing placeholder for item keys (e.g. {'audio': '/path/to/{uri}.wav'}) Returns ------- protocol : Protocol Protocol instance .)rj)splitrh get_protocolr`)r,r`rjrd task_name protocol_namerfprotocols r!rnRegistry.get_protocol-sL*37**S// -$$]3(( M)  r c## URHOnURU5nUR5H'nURU5HnUSUSU3v M M) MQ g7f)Nrl)r+rh get_tasks get_protocols)r,rdrfrorps r!__iter__Registry.__iter__Ksc!^^M((7H%//1 %-%;%;I%FM*O1YKqHH&G2,sA!A#rZc [U5n[5nUR5H@upgUR5H'up[U5n[XXU5n U cM"XXh4'M) MB XR;a&URUR n [ X[XAU5 UR5V V Vs/sH uupoX4PM nn n n[U[4[U5US.5URU'gs snn n f)aKLoad all protocols from this database into the registry. Parameters ---------- db_name : _type_ Name of the database db_entries : dict Dict of all entries under this database (this should be tasks) database_yml : Union[Text, Path], optional Path to the database.yml file. Not required for X protocols, by default None N)r- _protocols) rRr(rOrr+ry_merge_protocols_inplacetyperr)r,rYrZr9r0rWro task_entriesrqprotocol_entriesCustomProtocol old_protocolstaskp_namep_type protocol_lists r!rPRegistry._load_protocolsRs&g,486 '1'7'7'9 #I.:.@.@.B*x=!0L"")3A9/0/C(: nn $ NN73>>M $$  BKAR AR'=~v6 "AR  #'  K!-0 J# w s3C5c URRSS5 URR5HJupUR S[ 55nSU;dM'UR SUSU[RS9 ML g)z>Reloads all meta protocols from all database.yml files loaded.r?Nr>r@) r+rIr)rOrNr(rPrr)r,db_ymlrTr+s r!r6Registry._reload_meta_protocolssq 3%"ll002NF ;7Ii$$3k6J6J%3r )r)r+r*)r%Nr')rrrr__doc__r-rrr r rr7rr5rr4rrhrr rrnrvr(rPr6rrr r!r#r#gs62v(00&D$J&&6(,, U G,D$J'G,G,T G,R!"x!"H>B#+M#: <I+/'00 2 2 D$J' 2  2 h r r#r%c[RRSS5n/nURS5HDn[ U5R 5nUR 5(dM3URU5 MF U$)zParse PYANNOTE_DATABASE_CONFIG environment variable PYANNOTE_DATABASE_CONFIG may contain multiple paths separation by ";". Returns ------- paths : list of Path List of all YAML database file defined in PYANNOTE_DATABASE_CONF PYANNOTE_DATABASE_CONFIG;)osenvironrNrmrrBis_filerQ)contentpathsr/s r!_env_config_pathsrsajjnn7> LL # Lr c/n[S5R5nUR5(aURU5 [R"5S- nUR5(aURU5 U[ 5- nU$)zGet paths to default YAML configuration files * $HOME/.pyannote/database.yml * $CWD/database.yml * PYANNOTE_DATABASE_CONFIG environment variable Returns ------- paths : list of Path List of existing default YAML configuration files z~/.pyannote/database.ymlz database.yml)rrBrrQcwdr)r home_db_yml cwd_db_ymls r!_find_default_ymlsrsvE12==?K [!n,J Z    E Lr new_protocolsrr0rYr9cUR5HupVXP;aUupxUSUSU3n U[R:Xa[SU SUS35eU[R:Xa[ R "SU SUS35 MqU[R:Xa#[ R "SU S US35 X`U'MMX`U'M g ) aMerge new and old protocols inplace into the passed new_protocol. Warning, merging order might be counterintuitive : "KEEP" strategy keeps element from the OLD protocol and MODIFIES the new protocol. TODO: make it intuitive :) Parameters ---------- new_protocols : Dict[Tuple[Text, Text], Type] New protocol dict Maps (task,protocol) tuples to custom protocol classes old_protocols : Dict[Tuple[Text, Text], Type] Old protocols dict Maps (task,protocol) tuples to custom protocol classes. mode : LoadingMode How to handle override db_name : _type_ Name of the database (for logging/warning purposes) database_yml : str Path of the database.yml file (for logging/warning purposes) rlz Cannot load z protocol from 'z' as it already exists.zReplacing existing z! protocol by the one defined in 'z'.z Skipping z protocol defined in 'N)rOrr RuntimeErrorrwarningswarnr) rrr0rYr9p_idold_pt_namerrealnames r!rzrzs>%**,   !NF!!F81VH5H{(((""8*,<\NJab --- )(3TUaTbbde))) z)? ~Mde',d# *#($ 9-r )!enumrrpathlibrtypingrrrrr r r r r#pyannote.database.protocol.protocolr rcustomrrrfrrFrr#rrrRrzregistryymlr7rr r!rs> FFFG- $vkk\ 4:*DJ8;(dDj)4/0;(dDj)4/0;( ;( ;(  ;(~ :  C 3 r