From 9818ebb630012f0c01fcc6c6db495179cc4930b4 Mon Sep 17 00:00:00 2001
From: Bekzhan Bekbolatuly <bekzhan@google.com>
Date: Wed, 19 Apr 2023 09:56:23 -0700
Subject: [PATCH] Internal change

PiperOrigin-RevId: 525476655
---
 .../tasks/cc/text/language_detector/language_detector_test.cc | 4 ++--
 .../tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/mediapipe/tasks/cc/text/language_detector/language_detector_test.cc b/mediapipe/tasks/cc/text/language_detector/language_detector_test.cc
index 92dc493e0..2e9a58409 100644
--- a/mediapipe/tasks/cc/text/language_detector/language_detector_test.cc
+++ b/mediapipe/tasks/cc/text/language_detector/language_detector_test.cc
@@ -32,7 +32,7 @@ limitations under the License.
 #include "mediapipe/framework/port/gtest.h"
 #include "mediapipe/framework/port/status_matchers.h"
 #include "mediapipe/tasks/cc/common.h"
-#include "tensorflow/lite/core/shims/cc/shims_test_util.h"
+#include "tensorflow/lite/test_util.h"
 
 namespace mediapipe::tasks::text::language_detector {
 namespace {
@@ -75,7 +75,7 @@ absl::Status MatchesLanguageDetectorResult(
 
 }  // namespace
 
-class LanguageDetectorTest : public tflite_shims::testing::Test {};
+class LanguageDetectorTest : public tflite::testing::Test {};
 
 TEST_F(LanguageDetectorTest, CreateFailsWithMissingModel) {
   auto options = std::make_unique<LanguageDetectorOptions>();
diff --git a/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc b/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc
index 062d0746d..87bc97274 100644
--- a/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc
+++ b/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc
@@ -38,7 +38,7 @@ limitations under the License.
 #include "mediapipe/tasks/cc/vision/core/image_processing_options.h"
 #include "mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_result.h"
 #include "mediapipe/tasks/cc/vision/utils/image_utils.h"
-#include "tensorflow/lite/core/shims/cc/shims_test_util.h"
+#include "tensorflow/lite/test_util.h"
 #include "util/tuple/dump_vars.h"
 
 namespace mediapipe {