Skip to content

Commit 3165ba3

Browse files
author
Daniel Flores
committed
remove debugging import
1 parent 7bed1a9 commit 3165ba3

File tree

2 files changed

+1
-6
lines changed

2 files changed

+1
-6
lines changed

src/torchcodec/_core/Encoder.cpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,3 @@
1-
#include <iostream>
2-
extern "C" {
3-
#include <libavutil/pixdesc.h>
4-
}
51
#include <sstream>
62

73
#include "src/torchcodec/_core/AVIOTensorContext.h"

test/test_ops.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1241,9 +1241,8 @@ def test_video_encoder_test_round_trip(self, tmp_path, format):
12411241
encoded_path = str(tmp_path / f"encoder_output.{format}")
12421242

12431243
source_frames = self.decode(str(asset.path)).data
1244-
frame_rate = 30
1244+
frame_rate = 30 # Frame rate is fixed with num frames decoded
12451245

1246-
# Use C++ encode_video_to_file function directly
12471246
encode_video_to_file(source_frames, frame_rate, encoded_path)
12481247

12491248
# Check that PSNR for decode(encode(samples)) is above 30

0 commit comments

Comments
 (0)