Remove unnecessary output check (#508)

NicolasHug · web-flow · commit a8299be5b152 · 2025-02-11T14:41:56.000Z
diff --git a/src/torchcodec/decoders/_core/VideoDecoderOps.cpp b/src/torchcodec/decoders/_core/VideoDecoderOps.cpp
@@ -237,11 +237,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
`@@ -237,11 +237,6 @@ OpsFrameOutput get_next_frame(at::Tensor& decoder) {`
`237`	`237`	`} catch (const VideoDecoder::EndOfFileException& e) {`
`238`	`238`	`C10_THROW_ERROR(IndexError, e.what());`
`239`	`239`	`}`
`240`		`- if (result.data.sizes().size() != 3) {`
`241`		`- throw std::runtime_error(`
`242`		`- "image_size is unexpected. Expected 3, got: " +`
`243`		`- std::to_string(result.data.sizes().size()));`
`244`		`- }`
`245`	`240`	`return makeOpsFrameOutput(result);`
`246`	`241`	`}`
`247`	`242`