iMddlmZmZmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlm Z ddl mZdd lZdd lZdd lmZdd lmZdd lmZGd dZy ))OptionalTextIOUnionDictAny)Path) OrderedDict) PipelineInput)PipelineOutput) Direction)FileLockN)Timeline) Annotation)Trialc eZdZdZdZedZejdZdZdZ dZ dZ d(d e e d e e d efd Zd ed efdZded efdZ d)de ed e e d e e d efdZdZded dfdZded dfdZedZ d*dede ede ed efdZded dfdZded efdZd+dZ 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+y),PipelinezBase tunable pipelineclt|_t|_t|_d|_y)NF)r _parameters _instantiated _pipelinestrainingselfs o/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pyannote/pipeline/pipeline.py__init__zPipeline.__init__0s.2=.9]0;} c|jSN) _trainingrs rrzPipeline.training>s ~~rcd||_|jjD] \}}||_yr)r ritemsr)rr_pipelines rrzPipeline.trainingBs,!??002KAx (H 3rc |jd}ttt|j |j S)NT)frozen) parametershashtuplesorted_flattenr")rr&s r__hash__zPipeline.__hash__Is9-E&v!6!?@AArcFd|jvr|jd}||vr||Sd|jvr|jd}||vr||Sd|jvr|jd}||vr||Sdjt|j|}t |)z(Advanced) attribute getterrrrz!'{}' object has no attribute '{}')__dict__formattype__name__AttributeError)rnamerrrmsgs r __getattr__zPipeline.__getattr__Ns dmm + MM/:M}$$T** DMM )-- 6K{""4(( 4== (|4Jz!!$''188d9L9LdSS!!rcddlm}fd}|jjd}|jjd}|jjd}t ||r)| d}t |||j||||<yt |t r)| d }t |||j||||<y| |vr||<ytj||y) 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] }|vs|= yr)dictsdr3s r remove_fromz)Pipeline.__setattr__..remove_fromrs19$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__fs ) mm'' 6  ))/: ]]&&|4  eY '"W%S))  }j A %K   eX &!V$S))  {M B$Jt    "t{':"'M$  4u-rc||jvr|j|=y||jvr|j|=y||jvr|j|=ytj ||yr)rrrr@ __delattr__)rr3s rrDzPipeline.__delattr__sd 4## #  & T'' '""4( T__ $%   tT *rr& instantiatedreturncddlm}|r|r d}t||rt|j}n\|rE|j j Dcic]\}}t||s||j!}}}nt|j }|jj D]:\}} | j||} | j D]\} } | ||d| <<|Scc}}w)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`.r&rE>) r=rI ValueErrordictrrr"r?rBr_flattened_parameters) rr&rErIr4paramsnp pipeline_namer$pipeline_paramsr3rBs rrNzPipeline._flattened_parameterss& & lHCS/ ! $,,-F '+'7'7'='='?'?tq!:aQWCX177 '?  $**+F(,'<'<'> #M8&<<L=O /446 e49-$01 7 (? s C%$C% nested_paramsct}|jD]L\}}t|tr2|j|jD]\}}|||d|<H|||<N|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` rK)rMr"r?r+)rrTflattened_paramsr3rBsubnamesubvalues rr+zPipeline._flattensz: 6(..0KD%%&)-u)=)C)C)E%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` rKr rN)rr"splitlenjoin _unflatten) rrVrTr3rSrBtokensrR param_namer$s rr]zPipeline._unflattens< 04@48@+113KD%ZZ_F6{Q &q  XXfQRj1 =B .z: ', d#4$#oo335ND("*"5"5od6K"LM$ 6-As B6Ntrialc |s|r| d}t||j||}|*|jDcic]\}}||||}}}|j|Scc}}w)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. zKD%t&!%..tf5=>%S/)%,,U3t'''dF5M2v%56CS/ !'** rcddlm}|jD]\}}||jvr@t |t sd|d}t ||j|j|T||jvrlt||}t ||rF|j|k7r7d|d|jd|d}tj||j}t|||d |d }t ||j|S) zRecursively instantiate all pipelines Parameters ---------- params : `dict` Nested dictionary of parameters. Returns ------- self : `Pipeline` Instantiated pipeline. r rHrfz7' pipeline can be instantiated (not the whole pipeline)z Parameter 'z%' is frozen: using its frozen value (z) instead of the one provided (z).rgrh)r=rIr"rr?rMrL instantiatergetattrrBwarningswarnrjrd)rrOrIr3rBr4rbs rrlzPipeline.instantiates &!<<>KD%t&!%..tf5CD%S/)%11%8t'''d+eV,1E%dV,!KK=(GwbRMM#&!KKEdE*v%56CS/ !9*<  rct|j|j}t|j|jd}||k(S)zEWhether pipeline has been instantiated (and therefore can be applied)TrE)setr+r')rr'rEs rrEzPipeline.instantiatedsGt'89: 4==d)KLM \))r params_ymllossc8||jd}d|i}|||d<tj|d}t|j d5t |d 5}|j |dddddd|S#1swYxYw#1swY|SxYw) 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`. NTrqrOrtF)default_flow_stylez.lockwmode)r'yamldumpr with_suffixopenwrite)rrsrOrtcontent content_ymlfps r dump_paramszPipeline.dump_paramss. >__$_7FV$  "GFOiiEB j,,W5 6js+r%,7,+7s$BB1BB BBct|d5}tj|tj}ddd|j dS#1swYxYw)zInstantiate pipeline using parameters from disk Parameters ---------- param_yml : `Path` Path to YAML file. Returns ------- self : `Pipeline` Instantiated pipeline rrx)LoaderNrO)r}rzload SafeLoaderrl)rrsrrOs r load_paramszPipeline.load_paramssH*3 '2YYr$//:F(x 011( 's &AAinputct)z-Apply pipeline on input and return its outputNotImplementedError)rrs r__call__zPipeline.__call__s!!rct)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 "##rcy)Nminimizer9rs r get_directionzPipeline.get_directionsroutputct)aCompute loss for given input/output pair Parameters ---------- input : object Pipeline input. output : object Pipeline output Returns ------- loss : `float` Loss value r)rrrs rrtz Pipeline.losss "##rcy)Nrttmr9rs r write_formatzPipeline.write_format(srfilecBt|d|j||S)zWrite pipeline output to file Parameters ---------- file : file object output : object Pipeline output write_)rmr)rrrs rr~zPipeline.write,s(;wtvd&7&7%89:4HHrc ht|tr|jd}t|tr\|j dD]F\}}}d|j d|j dd|jdd |d }|j|Hy d |jjd }t|)zWrite pipeline output to "rttm" file Parameters ---------- file : file object output : `pyannote.core.Timeline` or `pyannote.core.Annotation` Pipeline output string) generatorT yield_labelzSPEAKER z 1 .3f z 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 )!,,,>1avzzl#aggc]!AJJs;KL!!"<1 4 ? v''0012  ! "#&&rc t|trD|D]>}|jd|jdd|jdd}|j |@yt|t r^|jdD]H\}}}|jd|jdd|jdd|d|d }|j |Jyd|jjd}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?rrrendr~rrrr1r)rrrrrrrr4s r write_txtzPipeline.write_txtTs fh ' **Qqwwsm1QUU3KrB 4   fj )!,,,>1a **Qqwwsm1QUU3Kq1QCrJ 4 ? v''0012  ! "#&&r)FF)NFF)NN)rFz pyannote.metrics.base.BaseMetric),r1 __module__ __qualname____doc__rpropertyrsetterr,r5rArDrboolrMrNr+r]rr'rdrirlrErfloatstrrrr r rrr rrtrrr~rrrrrr9rrrr-s __)) B "0/.b +NS-tn-rs::65#!" "$A 'A 'r