No public description

PiperOrigin-RevId: 611557924
tensorflow · Feb 29, 2024 · d8d1544 · d8d1544
1 parent 37e8e00
commit d8d1544
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 8 deletions.
diff --git a/tensorflow_text/core/kernels/sentencepiece/BUILD b/tensorflow_text/core/kernels/sentencepiece/BUILD
@@ -250,7 +250,6 @@ cc_test(
         "//file/base:path",
         "//file/localfile",
         "@com_google_googletest//:gtest_main",
-        "@com_google_absl//absl/flags:flag",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/strings:str_format",
         "@com_google_sentencepiece//:sentencepiece_cc_proto",
@@ -274,7 +273,6 @@ cc_test(
         "//file/base:path",
         "//file/localfile",
         "@com_google_googletest//:gtest_main",
-        "@com_google_absl//absl/flags:flag",
         "@com_google_absl//absl/strings:str_format",
         "@com_google_sentencepiece//:sentencepiece_cc_proto",
         "@com_google_sentencepiece//:sentencepiece_processor",

diff --git a/tensorflow_text/core/kernels/sentencepiece/optimized_decoder_test.cc b/tensorflow_text/core/kernels/sentencepiece/optimized_decoder_test.cc
@@ -34,7 +34,6 @@ limitations under the License.
 #include "file/base/path.h"
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>
-#include "absl/flags/flag.h"
 #include "absl/strings/str_format.h"
 #include "src/sentencepiece.proto.h"
 #include "src/sentencepiece_processor.h"
@@ -80,8 +79,7 @@ TEST(OptimizedEncoder, ConfigConverter) {
   std::string config;
 
   auto status = internal::TFReadFileToString(
-      file::JoinPath(absl::GetFlag(FLAGS_test_srcdir), kConfigFilePath),
-      &config);
+      file::JoinPath(::testing::SrcDir(), kConfigFilePath), &config);
   ASSERT_TRUE(status.ok());
 
   ::sentencepiece::SentencePieceProcessor processor;

diff --git a/tensorflow_text/core/kernels/sentencepiece/optimized_encoder_test.cc b/tensorflow_text/core/kernels/sentencepiece/optimized_encoder_test.cc
@@ -34,7 +34,6 @@ limitations under the License.
 #include "file/base/path.h"
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>
-#include "absl/flags/flag.h"
 #include "absl/status/status.h"
 #include "absl/strings/str_format.h"
 #include "src/sentencepiece.proto.h"
@@ -162,8 +161,7 @@ TEST(OptimizedEncoder, NormalizeStringWhitespacesRemove) {
 TEST(OptimizedEncoder, ConfigConverter) {
   std::string config;
   auto status = internal::TFReadFileToString(
-      file::JoinPath(absl::GetFlag(FLAGS_test_srcdir), kConfigFilePath),
-      &config);
+      file::JoinPath(::testing::SrcDir(), kConfigFilePath), &config);
   ASSERT_TRUE(status.ok());
 
   ::sentencepiece::SentencePieceProcessor processor;