From 76778ac8c66025546842ecf7dd17ad8b15e7a155 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Thu, 13 Nov 2025 16:59:30 +0000 Subject: [PATCH] Attempt at fixing contiguity test failures --- test/test_encoders.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/test/test_encoders.py b/test/test_encoders.py index 0a360ccf9..a90d6a941 100644 --- a/test/test_encoders.py +++ b/test/test_encoders.py @@ -708,17 +708,20 @@ def test_contiguity(self, method, tmp_path): ) def encode_to_tensor(frames): + common_params = dict(crf=0, pixel_format="yuv444p") if method == "to_file": dest = str(tmp_path / "output.mp4") - VideoEncoder(frames, frame_rate=30).to_file(dest=dest) + VideoEncoder(frames, frame_rate=30).to_file(dest=dest, **common_params) with open(dest, "rb") as f: return torch.frombuffer(f.read(), dtype=torch.uint8) elif method == "to_tensor": - return VideoEncoder(frames, frame_rate=30).to_tensor(format="mp4") + return VideoEncoder(frames, frame_rate=30).to_tensor( + format="mp4", **common_params + ) elif method == "to_file_like": file_like = io.BytesIO() VideoEncoder(frames, frame_rate=30).to_file_like( - file_like, format="mp4" + file_like, format="mp4", **common_params ) return torch.frombuffer(file_like.getvalue(), dtype=torch.uint8) else: