o i @sUddlmZddlmZeeeeBZdededefddZdededefdd Z d ededefd d Z eeegefZ ee e d Z e ee fed<ddZdedededefddZdZdS))Callable) FrameBatchvalues desired_lenreturncCs||dg|t|7}|S)Nlenrrr Z/home/ubuntu/Irodori-TTS/.venv/lib/python3.10/site-packages/torchcodec/samplers/_common.py_repeat_last_policysr cCs||t|dd|S)Nrr r r r _wrap_policysrframes_indicescCstd)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)rrr r r _error_policysr) repeat_lastwraperror_POLICY_FUNCTIONScCsbt|dkrtdt|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_clippolicyr r r _validate_common_params,s   rframes num_clipsrcCsD|jjdd}t|jj||g|R|j|||j||dS)N)data pts_secondsduration_seconds)r shaperviewr!r")rrr last_3_dimsr r r _reshape_4d_framebatch_into_5d<s   r&a 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)collections.abcr torchcodecrlistintfloat_LIST_OF_INT_OR_FLOATr rr_POLICY_FUNCTION_TYPErdictstr__annotations__rr&_FRAMEBATCH_RETURN_DOCSr r r r sP