Automated Code Change

PiperOrigin-RevId: 610211777
tensorflow · Feb 25, 2024 · 704e250 · 704e250
1 parent 27923d3
commit 704e250
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/tensorflow_serving/util/prometheus_exporter.cc b/tensorflow_serving/util/prometheus_exporter.cc
@@ -162,7 +162,7 @@ PrometheusExporter::PrometheusExporter()
 Status PrometheusExporter::GeneratePage(string* http_page) {
   if (http_page == nullptr) {
     return Status(
-        static_cast<tsl::errors::Code>(absl::StatusCode::kInvalidArgument),
+        static_cast<absl::StatusCode>(absl::StatusCode::kInvalidArgument),
         "Http page pointer is null");
   }
   monitoring::CollectionRegistry::CollectMetricsOptions collect_options;
@@ -186,7 +186,7 @@ Status PrometheusExporter::GeneratePage(string* http_page) {
   }
   *http_page = absl::StrJoin(lines, "\n");
   absl::StrAppend(http_page, "\n");
-  return OkStatus();
+  return absl::OkStatus();
 }
 
 }  // namespace serving

diff --git a/tensorflow_serving/util/status_util.cc b/tensorflow_serving/util/status_util.cc
@@ -33,7 +33,7 @@ StatusProto ToStatusProto(const Status& status) {
 Status FromStatusProto(const StatusProto& status_proto) {
   return status_proto.error_code() == tensorflow::error::OK
              ? Status()
-             : Status(static_cast<tsl::errors::Code>(status_proto.error_code()),
+             : Status(static_cast<absl::StatusCode>(status_proto.error_code()),
                       status_proto.error_message());
 }
 

diff --git a/tensorflow_serving/util/status_util_test.cc b/tensorflow_serving/util/status_util_test.cc
@@ -23,7 +23,7 @@ namespace {
 
 TEST(StatusUtilTest, ConvertsErrorStatusToStatusProto) {
   Status status =
-      Status(static_cast<tsl::errors::Code>(absl::StatusCode::kAborted),
+      Status(static_cast<absl::StatusCode>(absl::StatusCode::kAborted),
              "aborted error message");
   StatusProto status_proto = ToStatusProto(status);
   EXPECT_EQ(tensorflow::error::ABORTED, status_proto.error_code());