Merge branch 'main' into addVideoStream

NicolasHug · web-flow · commit c5e1bc38ffb1 · 2025-02-11T14:42:07.000Z
diff --git a/src/torchcodec/decoders/_core/VideoDecoderOps.cpp b/src/torchcodec/decoders/_core/VideoDecoderOps.cpp
@@ -236,11 +236,6 @@ OpsFrameOutput get_next_frame(at::Tensor& decoder) {
   } catch (const VideoDecoder::EndOfFileException& e) {
     C10_THROW_ERROR(IndexError, e.what());
   }
-  if (result.data.sizes().size() != 3) {
-    throw std::runtime_error(
-        "image_size is unexpected. Expected 3, got: " +
-        std::to_string(result.data.sizes().size()));
-  }
   return makeOpsFrameOutput(result);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -236,11 +236,6 @@ OpsFrameOutput get_next_frame(at::Tensor& decoder) {`
`236`	`236`	`} catch (const VideoDecoder::EndOfFileException& e) {`
`237`	`237`	`C10_THROW_ERROR(IndexError, e.what());`
`238`	`238`	`}`
`239`		`- if (result.data.sizes().size() != 3) {`
`240`		`- throw std::runtime_error(`
`241`		`- "image_size is unexpected. Expected 3, got: " +`
`242`		`- std::to_string(result.data.sizes().size()));`
`243`		`- }`
`244`	`239`	`return makeOpsFrameOutput(result);`
`245`	`240`	`}`
`246`	`241`