From 53e4e925052f017b2f88eb69a12356e9e1636d49 Mon Sep 17 00:00:00 2001
From: MediaPipe Team <mediapipe-team@google.com>
Date: Fri, 24 Mar 2023 09:04:53 -0700
Subject: [PATCH] Support single channel golden images.

PiperOrigin-RevId: 519158051
---
 mediapipe/framework/tool/test_util.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mediapipe/framework/tool/test_util.cc b/mediapipe/framework/tool/test_util.cc
index c7ed063e0..e8b02084b 100644
--- a/mediapipe/framework/tool/test_util.cc
+++ b/mediapipe/framework/tool/test_util.cc
@@ -214,7 +214,8 @@ absl::Status CompareAndSaveImageOutput(
     const ImageFrameComparisonOptions& options) {
   ASSIGN_OR_RETURN(auto output_img_path, SavePngTestOutput(actual, "output"));
 
-  auto expected = LoadTestImage(GetTestFilePath(golden_image_path));
+  auto expected =
+      LoadTestImage(GetTestFilePath(golden_image_path), ImageFormat::UNKNOWN);
   if (!expected.ok()) {
     return expected.status();
   }