Vji UddlmZmZddlmZeeeeefZdededefdZ dededefdZ dededefd Z eeegefZ e e e d Z eee fed <d Zd edededefdZdZdS))CallableUnion) FrameBatchvalues desired_lenreturncD||dg|t|z zz }|S)Nlenrrs U/root/voice-cloning/.venv/lib/python3.11/site-packages/torchcodec/samplers/_common.py_repeat_last_policyrs)  vbzlkCKK788F McB||t|zdzzd|S)Nr r s r _wrap_policyrs( kS[[014 5| | DDrframes_indicesc td)NzYou set the 'error' policy, and the sampler tried to decode a frame that is beyond the number of frames in the video. Try to leave sampling_range_end to its default value?) ValueError)rrs r _error_policyrs  @  r) repeat_lastwraperror_POLICY_FUNCTIONSc.t|dkr tdt|d|dkrtd|d|tvr-td|dtd dS) Nrz,Decoder must have at least one frame, found z frames.rznum_frames_per_clip (z) must be strictly positivezInvalid policy (z). Supported values are .)r rrkeys)decodernum_frames_per_clippolicys r_validate_common_paramsr",s 7||a Q3w<< Q Q Q   a T$7 T T T   &++---- Zv Z Z?P?U?U?W?W Z Z Z   .-rframes num_clipsr c|jjdd}t|jj||g|R|j|||j||S)N)data pts_secondsduration_seconds)r'shaperviewr(r))r#r$r last_3_dimss r_reshape_4d_framebatch_into_5dr-<sy +#BCC(K  V[ i)< K{ K K K&++I7JKK055iATUU   ra Returns: FrameBatch: The sampled :term:`clips`, as a 5D :class:`~torchcodec.FrameBatch`. The shape of the ``data`` field is (``num_clips``, ``num_frames_per_clips``, ...) where ... is (H, W, C) or (C, H, W) depending on the ``dimension_order`` parameter of :class:`~torchcodec.decoders.VideoDecoder`. The shape of the ``pts_seconds`` and ``duration_seconds`` fields is (``num_clips``, ``num_frames_per_clips``). N)typingrr torchcodecrlistintfloat_LIST_OF_INT_OR_FLOATrrr_POLICY_FUNCTION_TYPErdictstr__annotations__r"r-_FRAMEBATCH_RETURN_DOCSrrr:sy"""""""""!!!!!!d3ie45 !03E !E03EEEEE)8;!"7!=?T!TU'  774223              r