o %ݫi @s&dZddlZddZGdddZdS)z=Chaining together callables, if some require relative lengthsNcCst|}d|j|jvS)aCheck if func takes ``lengths`` keyword argument. Arguments --------- func : callable The function, method, or other callable to search for the lengths arg. Returns ------- True if func takes ``lengths`` keyword argument. lengths)inspectgetfullargspecargs kwonlyargs)funcspecr P/home/ubuntu/.local/lib/python3.10/site-packages/speechbrain/utils/callchains.pylengths_arg_existss r c@s2eZdZdZddZd ddZddZd d ZdS) LengthsCapableChaina2Chain together callables. Can handle relative lengths. This is a more light-weight version of speechbrain.nnet.containers.LengthsCapableSequential Arguments --------- *funcs : list, optional Any number of functions or other callables, given in order of execution. cGs$g|_g|_|D]}||qdSN)funcs takes_lengthsappend)selfrrr r r __init__#s  zLengthsCapableChain.__init__NcCsP|js|St|j|jD]\}}|r|||}n||}t|tr%|d}q |S)aRun the chain of callables on the given input Arguments --------- x : Any The main input lengths : Any The lengths argument which will be conditionally passed to any functions in the chain that take a 'lengths' argument. In SpeechBrain the convention is to use relative lengths. Returns ------- The input as processed by each function. If no functions were given, simply returns the input. Note ---- By convention, if a callable in the chain returns multiple outputs (returns a tuple), only the first output is passed to the next callable in the chain. r)rzipr isinstancetuple)rxrr give_lengthsr r r __call__)s  zLengthsCapableChain.__call__cCs |j||jt|dS)zAdd a function to the chainN)rrrr )rrr r r rKs zLengthsCapableChain.appendcCs8|jj}|jr|dddd|jDSd|S)Nz:  css|]}t|VqdSr )str).0fr r r Ssz.LengthsCapableChain.__str__..zEmpty ) __class____name__rjoin)rclsnamer r r __str__Ps  zLengthsCapableChain.__str__r )r __module__ __qualname____doc__rrrr"r r r r r s  " r )r%rr r r r r r s