i UddlmZddlmZeeeezZdededefdZdededefdZ dededefd Z eeegefZ ee e d Z e ee fed <d Zd edededefdZdZy))Callable) FrameBatchvalues desired_lenreturnc6||dg|t|z zz }|S)Nlenrrs p/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torchcodec/samplers/_common.py_repeat_last_policyrs'  vbzlkCK788F Mc0||t|zdzzd|S)Nr r s r _wrap_policyrs$ kS[014 5| DDrframes_indicesctd)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_FUNCTIONSct|dkrtdt|d|dkrtd|d|tjvr$td|dtjd y) 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:3w<. Q  a#$7#88S T  &++--vh&>?P?U?U?W>XXY Z  .rframes num_clipsrc|jjdd}t|jj||g||jj|||j j||S)N)data pts_secondsduration_seconds)r&shaperviewr'r()r"r#r last_3_dimss r _reshape_4d_framebatch_into_5dr,<sq ++##BC(K  V[[  i)< K{ K&&++I7JK0055iATU 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)collections.abcr torchcodecrlistintfloat_LIST_OF_INT_OR_FLOATrrr_POLICY_FUNCTION_TYPErdictstr__annotations__r!r,_FRAMEBATCH_RETURN_DOCSrr r9s$!S DK/ !03E !E03EE)8;!"7!=?T!TU'  74223         r