jMddlmZmZmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlm Z ddl mZdd lZdd lZdd lmZdd lmZdd lmZGd dZd S))OptionalTextIOUnionDictAny)Path) OrderedDict) PipelineInput)PipelineOutput) Direction)FileLockN)Timeline) Annotation)Trialc eZdZdZdZedZejdZdZdZ dZ dZ d*d e e d e e d efd Zded efdZded efdZ d+de ed e e d e e d efdZdZded dfdZded dfdZedZ d,dede ede ed efdZded dfdZded efdZd-d!Z d e!fd"Z"ded#ed efd$Z#ed%Z$d&e%d#efd'Z&d&e%d#e'e(e)ffd(Z*d&e%d#e'e(e)ffd)Z+dS).PipelinezBase tunable pipelinect|_t|_t|_d|_dS)NF)r _parameters _instantiated _pipelinestrainingselfs }/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/pyannote/pipeline/pipeline.py__init__zPipeline.__init__0s82=.9]]0;}} c|jSN) _trainingrs rrzPipeline.training>s ~rc`||_|jD] \}}||_ dSr)r ritemsr)rr_pipelines rrzPipeline.trainingBs?!?0022 ) )KAx (H   ) )rc |d}ttt||S)NT)frozen) parametershashtuplesorted_flattenr")rr&s r__hash__zPipeline.__hash__IsK--E&v!6!6!!>??@@AAArc<d|jvr|jd}||vr||Sd|jvr|jd}||vr||Sd|jvr|jd}||vr||Sdt|j|}t |)z(Advanced) attribute getterrrrz!'{}' object has no attribute '{}')__dict__formattype__name__AttributeError)rnamerrrmsgs r __getattr__zPipeline.__getattr__Ns dm + + M/:M}$$$T** DM ) )- 6K{"""4(( 4= ( (|4Jz!!!$''188d9LdSSS!!!rcddlm}fd}|jd}|jd}|jd}t ||r,|d}t |||j||||<dSt |t r,|d }t |||j||||<dS| |vr||<dSt||dS) a?(Advanced) attribute setter If `value` is an instance of `Parameter`, store it in `_parameters`. elif `value` is an instance of `Pipeline`, store it in `_pipelines`. elif `value` isn't an instance of `Parameter` and `name` is in `_parameters`, store `value` in `_instantiated`. r ) Parameterc |D] }|vr|= dSr)dictsdr3s r remove_fromz)Pipeline.__setattr__..remove_fromrs-  199$  rrrrNz>cannot assign hyper-parameters before Pipeline.__init__() callz;cannot assign sub-pipelines before Pipeline.__init__() call) parameterr7r.get isinstancer2robject __setattr__) rr3valuer7r<rrrr4s ` rrAzPipeline.__setattr__fsY )(((((     m'' 66  ))/:: ]&&|44  eY ' ' "W%S))) K }j A A A %K  F eX & & !V$S))) K {M B B B$Jt  F  "t{':':"'M$  F4u-----rc||jvr |j|=dS||jvr |j|=dS||jvr |j|=dSt||dSr)rrrr@ __delattr__)rr3s rrDzPipeline.__delattr__s{ 4# # # &&& T' ' '"4((( T_ $ $%%%   tT * * * * *rFr& instantiatedreturnc ddlm |r|rd}t||rt|j}n<|r& fd|jD}nt|j}|jD]@\}}|||}|D]\}} | ||d|<A|S)awGet flattened dictionary of parameters Parameters ---------- frozen : `bool`, optional Only return value of frozen parameters. instantiated : `bool`, optional Only return value of instantiated parameters. Returns ------- params : `dict` Flattened dictionary of parameters. r Frozenz4one must choose between `frozen` and `instantiated`.cFi|]\}}t|||jSr9)r?rB).0nprIs r z2Pipeline._flattened_parameters..sA#q!:aQWCXCX17rr&rE>) r=rI ValueErrordictrrr"r_flattened_parameters) rr&rEr4params pipeline_namer$pipeline_paramsr3rBrIs @rrSzPipeline._flattened_parameterss1& &%%%%%  "l "HCS// !  ,$,--FF  ,'+'7'='='?'?FF $*++F(,'<'<'>'> : : #M8&<<L=O /4466 : : e49-00$0011 : r nested_paramsct}|D]W\}}t|tr8||D]\}}|||d|<R|||<X|S)aConvert nested dictionary to flattened dictionary For instance, a nested dictionary like this one: ~~~~~~~~~~~~~~~~~~~~~ param: value1 pipeline: param: value2 subpipeline: param: value3 ~~~~~~~~~~~~~~~~~~~~~ becomes the following flattened dictionary: ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ param : value1 pipeline>param : value2 pipeline>subpipeline>param : value3 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Parameter --------- nested_params : `dict` Returns ------- flattened_params : `dict` rP)rRr"r?r+)rrWflattened_paramsr3rBsubnamesubvalues rr+zPipeline._flattens: 66(..00 / /KD%%&& /)-u)=)=)C)C)E)EEE%GXparam : value2 pipeline>subpipeline>param : value3 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ becomes the following nested dictionary: ~~~~~~~~~~~~~~~~~~~~~ param: value1 pipeline: param: value2 subpipeline: param: value3 ~~~~~~~~~~~~~~~~~~~~~ Parameter --------- flattened_params : `dict` Returns ------- nested_params : `dict` ci|]}|iSr9r9)rKr3s rrNz'Pipeline._unflatten..s@@@4@@@rrPr rN)rr"splitlenjoin _unflatten) rrYrWrVr3rBtokensrU param_namer$s rrazPipeline._unflattens< @@@@@+1133 , ,KD%ZZ__F6{{Q &q  XXfQRRj11 =B .z:: ', d###o3355 M MND("*"5"5od6K"L"LM$  rNtrialc|s|rd}t||||} fd|D}||S)agReturns nested dictionary of (optionnaly instantiated) parameters. For a pipeline with one `param`, one sub-pipeline with its own param and its own sub-pipeline, it will returns something like: ~~~~~~~~~~~~~~~~~~~~~ param: value1 pipeline: param: value2 subpipeline: param: value3 ~~~~~~~~~~~~~~~~~~~~~ Parameter --------- trial : `Trial`, optional When provided, use trial to suggest new parameter values and return them. frozen : `bool`, optional Return frozen parameter value instantiated : `bool`, optional Return instantiated parameter values. Returns ------- params : `dict` Nested dictionary of parameters. See above for the actual format. Nz.^s+QQQ;4dEE$..QQQr)rQrSr"ra)rrdr&rEr4rTs ` rr'zPipeline.parameters2sF  "F "(9PCS// !++6 +UU  QQQQ&,,..QQQFv&&&rcdS)z8Instantiate root pipeline with current set of parametersNr9rs r initializezPipeline.initializecs rrTc^ddlm}|D]\}}||jvrKt |t sd|d}t ||j||Y||jvrt||||}d|d}t ||S)zRecursively freeze pipeline parameters Parameters ---------- params : `dict` Nested dictionary of parameters. Returns ------- self : `Pipeline` Pipeline. r rHonly parameters of 'z1' pipeline can be frozen (not the whole pipeline) parameter '' does not exist) r=rIr"rr?rRrQfreezersetattr)rrTrIr3rBr4s rrnzPipeline.freezegs &%%%%%!<<>> " "KD%t&&!%..*>t>>>%S//)%,,U333t'''dFF5MM2226666CS// ! rc"ddlm}|D]\}}||jvrKt |t sd|d}t ||j||Y||jvrit||}t ||r7|j |kr,d|d|j d|d}tj ||j }t|||d |d }t |||S) zRecursively instantiate all pipelines Parameters ---------- params : `dict` Nested dictionary of parameters. Returns ------- self : `Pipeline` Instantiated pipeline. r rHrkz7' pipeline can be instantiated (not the whole pipeline)z Parameter 'z%' is frozen: using its frozen value (z) instead of the one provided (z).rlrm)r=rIr"rr?rRrQ instantiatergetattrrBwarningswarnrori)rrTrIr3rBr4rgs rrqzPipeline.instantiatesx &%%%%%!<<>> " "KD%t&&!%..*DtDDD%S//)%11%888t'''d++eV,,(1E1ERdRR!KRRHMRRRM#&&&!KEdE***6666CS// !  rct||}t||d}||kS)zEWhether pipeline has been instantiated (and therefore can be applied)TrE)setr+r')rr'rEs rrEzPipeline.instantiatedsYt'8'899:: 4==d)K)KLLMM \))r params_ymllossch||d}d|i}|||d<tj|d}t|d5t |d 5}||dddn #1swxYwYdddn #1swxYwY|S) aDump parameters to disk Parameters ---------- params_yml : `Path` Path to YAML file. params : `dict`, optional Nested Parameters. Defaults to pipeline current parameters. loss : `float`, optional Loss value. Defaults to not write loss to file. Returns ------- content : `str` Content written in `param_yml`. NTrvrTryF)default_flow_stylez.lockwmode)r'yamldumpr with_suffixopenwrite)rrxrTrycontent content_ymlfps r dump_paramszPipeline.dump_paramss=. >__$_77FV$  "GFOiEBBB j,,W55 6 6 & &js+++ &r%%% & & & & & & & & & & & & & & & & & & & & & & & & & & & & & &s6B'.B B'B B'B B''B+.B+ct|d5}tj|tj}dddn #1swxYwY||dS)zInstantiate pipeline using parameters from disk Parameters ---------- param_yml : `Path` Path to YAML file. Returns ------- self : `Pipeline` Instantiated pipeline rr})LoaderNrT)rrload SafeLoaderrq)rrxrrTs r load_paramszPipeline.load_paramss*3 ' ' ' ;2Yr$/:::F ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;x 0111s!?AAinputct)z-Apply pipeline on input and return its outputNotImplementedError)rrs r__call__zPipeline.__call__s!!r pyannote.metrics.base.BaseMetricct)aReturn new metric (from pyannote.metrics) When this method is implemented, the returned metric is used as a replacement for the loss method below. Returns ------- metric : `pyannote.metrics.base.BaseMetric` rrs r get_metriczPipeline.get_metrics"###rcdS)Nminimizer9rs r get_directionzPipeline.get_directionszroutputct)aCompute loss for given input/output pair Parameters ---------- input : object Pipeline input. output : object Pipeline output Returns ------- loss : `float` Loss value r)rrrs rryz Pipeline.losss"###rcdS)Nrttmr9rs r write_formatzPipeline.write_format(svrfilecFt|d|j||S)zWrite pipeline output to file Parameters ---------- file : file object output : object Pipeline output write_)rrr)rrrs rrzPipeline.write,s,;wt9d&799::4HHHrc ht|tr|d}t|trT|dD];\}}}d|jd|jdd|jdd |d }|| z NDumping z, instances to "rttm" files is not supported.) r?r to_annotationr itertracksuristartdurationr __class__r1r)rrrstlliner4s r write_rttmzPipeline.write_rttm8s fh ' ' >))H)==F fj ) ) !,,,>> ! !1a1vz11agM11AJM11!"111 4    F !v'0 ! ! ! "#&&&rc t|tr8|D]3}|jd|jdd|jdd}||4dSt|t rV|dD]=\}}}|jd|jdd|jdd|d|d }||>dSd|jj d}t|) zWrite pipeline output to "txt" file Parameters ---------- file : file object output : `pyannote.core.Timeline` or `pyannote.core.Annotation` Pipeline output rr NTrrz+ instances to "txt" files is not supported.) r?rrrendrrrrr1r)rrrrrrrr4s r write_txtzPipeline.write_txtTs& fh ' '  ! ! *BBqwBBBQUBBBB 4    F fj ) ) !,,,>> ! !1a *JJqwJJJQUJJJJJQJJJ 4    F !v'0 ! ! ! "#&&&r)FF)NFF)NN)rFr),r1 __module__ __qualname____doc__rpropertyrsetterr,r5rArDrboolrRrSr+rarr'rirnrqrErfloatstrrrr r rrr rryrrrrrrrrr9rrrr-sv   X_))_) BBB """0/././.b + + +NS--tn-rs!:65555555555555######!!!!!!"""""" """"""$$$$$$A 'A 'A 'A 'A 'A 'A 'A 'A 'A 'r