o si @sLddlZddlmZddlZddZdddZd d Zdd d ZddZ dS)N)MutableMappingcCs8t|}|j|}|durdS|jtjjtjjfvS)a%Checks if a callable accepts a given keyword argument. Args: fn (callable): Callable to inspect. name (str): Check if ``fn`` can be called with ``name`` as a keyword argument. Returns: bool: whether ``fn`` accepts a ``name`` keyword argument. NF)inspect signature parametersgetkind ParameterPOSITIONAL_OR_KEYWORD KEYWORD_ONLY)fnnamer parameterrP/home/ubuntu/.local/lib/python3.10/site-packages/asteroid/utils/generic_utils.pyhas_args r_cCsdg}|D]'\}}|r|||n|}t|tr&|t|||dq|||fqt|S)aFlattens a dictionary into a single-level dictionary while preserving parent keys. Taken from `SO `_ Args: d (MutableMapping): Dictionary to be flattened. parent_key (str): String to use as a prefix to all subsequent keys. sep (str): String to use as a separator between two key levels. Returns: dict: Single-level dictionary, flattened. )sep)items isinstancerextend flatten_dictappenddict)d parent_keyrrkvnew_keyrrrrs  rcCs:t|}|D]\}}t|tjrt|||<q|S)zTake average of numpy arrays in a dictionary. Args: dic (dict): Input dictionary to take average from Returns: dict: New dictionary with array averaged. )rrrnpndarrayfloatmean)dicrrrrraverage_arrays_in_dic2s   r$}cCs>|durd|fStjdtd||}t|||}||fS)zGet indexes for a chunk of signal of a given length. Args: signal_len (int): length of the signal to trim. desired_len (int): the length of [start:stop] Returns: tuple: random start integer, stop integer. Nr)rrandomrandintmaxmin) signal_len desired_len rand_startstoprrrget_wav_random_start_stopDs  r/c CsNg}t|D]\}}}}}|r|r|}nd}|||||||fqt|S)aYGet list of decoder arguments for upsampling (right) side of a symmetric u-net, given the arguments used to construct the encoder. Args: encoders (tuple of length `N` of tuples of (in_chan, out_chan, kernel_size, stride, padding)): List of arguments used to construct the encoders skip_connections (bool): Whether to include skip connections in the calculation of decoder input channels. Return: tuple of length `N` of tuples of (in_chan, out_chan, kernel_size, stride, padding): Arguments to be used to construct decoders r)reversedrtuple) encodersskip_connections decoder_args enc_in_chan enc_out_chanenc_kernel_size enc_stride enc_padding skip_in_chanrrrunet_decoder_argsUsr;)rr)r%) rcollections.abcrnumpyrrrr$r/r;rrrrs