Skip to content

Commit 0443c3b

Browse files
committed
Merge branch 'main' of github.com:pytorch/torchcodec into audio_support
2 parents b572276 + 82224c4 commit 0443c3b

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

src/torchcodec/decoders/_video_decoder.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ class VideoDecoder:
2626
Args:
2727
source (str, ``Pathlib.path``, ``torch.Tensor``, or bytes): The source of the video.
2828
29-
- If ``str`` or ``Pathlib.path``: a path to a local video file.
29+
- If ``str``: a local path or a URL to a video file.
30+
- If ``Pathlib.path``: a path to a local video file.
3031
- If ``bytes`` object or ``torch.Tensor``: the raw encoded video data.
3132
stream_index (int, optional): Specifies which stream in the video to decode frames from.
3233
Note that this index is absolute across all media types. If left unspecified, then

test/decoders/test_video_decoder.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
cpu_and_cuda,
2020
get_ffmpeg_major_version,
2121
H265_VIDEO,
22+
in_fbcode,
2223
NASA_VIDEO,
2324
)
2425

@@ -876,6 +877,8 @@ def test_get_key_frame_indices(self, device):
876877
key_frame_indices, h265_reference_key_frame_indices, atol=0, rtol=0
877878
)
878879

880+
# TODO investigate why this fails internally.
881+
@pytest.mark.skipif(in_fbcode(), reason="Compile test fails internally.")
879882
@pytest.mark.parametrize("device", cpu_and_cuda())
880883
def test_compile(self, device):
881884
decoder = VideoDecoder(NASA_VIDEO.path, device=device)

0 commit comments

Comments
 (0)