Skip to content

Commit 920a36c

Browse files
committed
reduce decoded frames to avoid OOM in test_video_encoder_against_ffmpeg_cli
1 parent d12c895 commit 920a36c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_encoders.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -570,12 +570,12 @@ def write(self, data):
570570

571571
class TestVideoEncoder:
572572
def decode(self, source=None) -> torch.Tensor:
573-
return VideoDecoder(source).get_frames_in_range(start=0, stop=60).data
573+
return VideoDecoder(source).get_frames_in_range(start=0, stop=30).data
574574

575575
# TODO: add average_fps field to TestVideo asset
576576
def decode_and_get_frame_rate(self, source=None):
577577
decoder = VideoDecoder(source)
578-
frames = decoder.get_frames_in_range(start=0, stop=60).data
578+
frames = decoder.get_frames_in_range(start=0, stop=30).data
579579
frame_rate = decoder.metadata.average_fps
580580
return frames, frame_rate
581581

0 commit comments

Comments
 (0)