Support both RawIOBase and BytesIO

Author: scotts

src/torchcodec/decoders/_core/ops.py

@@ -137,7 +137,7 @@ def create_from_bytes(
 
 
 def create_from_file_like(
-    file_like: io.RawIOBase, seek_mode: Optional[str] = None
+    file_like: io.RawIOBase | io.BytesIO, seek_mode: Optional[str] = None
 ) -> torch.Tensor:
     assert _pybind_ops is not None
     return _convert_to_tensor(_pybind_ops.create_from_file_like(file_like, seek_mode))

test/decoders/test_ops.py

@@ -343,7 +343,10 @@ def get_frame1_and_frame_time6(decoder):
         assert_frames_equal(frame_time6, reference_frame_time6.to(device))
 
     @pytest.mark.parametrize("device", cpu_and_cuda())
-    @pytest.mark.parametrize("create_from", ("file", "tensor", "bytes", "file_like"))
+    @pytest.mark.parametrize(
+        "create_from",
+        ("file", "tensor", "bytes", "file_like_rawio", "file_like_bufferedio"),
+    )
     def test_create_decoder(self, create_from, device):
         path = str(NASA_VIDEO.path)
         if create_from == "file":
@@ -356,8 +359,12 @@ def test_create_decoder(self, create_from, device):
             with open(path, "rb") as f:
                 video_bytes = f.read()
             decoder = create_from_bytes(video_bytes)
-        elif create_from == "file_like":
+        elif create_from == "file_like_rawio":
             decoder = create_from_file_like(open(path, mode="rb", buffering=0), "exact")
+        elif create_from == "file_like_bufferedio":
+            decoder = create_from_file_like(
+                open(path, mode="rb", buffering=-4096), "exact"
+            )
         else:
             raise ValueError("Oops, double check the parametrization of this test!")