Skip to content

Commit 550d8c3

Browse files
committed
Merge branch 'main' of github.com:meta-pytorch/torchcodec into docssss
2 parents d805b5d + 4b32769 commit 550d8c3

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

test/smoke_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -374,8 +374,8 @@ def test_cuda_encoding(self, tmp_path):
374374
width=WIDTH,
375375
frame_rate=FRAME_RATE,
376376
)
377-
with cpu_enc.open(dest=cpu_path):
378-
cpu_video.write(frames.cpu())
377+
with cpu_enc.open_file(dest=cpu_path):
378+
cpu_video.add_frames(frames.cpu())
379379

380380
cuda_decoded = VideoDecoder(cuda_path).get_all_frames()
381381
cpu_decoded = VideoDecoder(cpu_path).get_all_frames()

test/test_encoders.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1863,6 +1863,7 @@ def test_device_cuda_0_string(self, tmp_path, method):
18631863
assert decoded_frames.shape == source_frames.shape
18641864

18651865
@pytest.mark.needs_cuda
1866+
@pytest.mark.parametrize("method", ("to_file", "to_file_like"))
18661867
def test_write_samples_on_cuda_errors(self, tmp_path, method):
18671868
enc, _, open_kwargs = self._create_encoder(method, tmp_path, "wav")
18681869
audio = enc.add_audio(sample_rate=44100, num_channels=1)

0 commit comments

Comments
 (0)