Automated Code Change

PiperOrigin-RevId: 690971099
tensorflow · Oct 29, 2024 · 96050a2 · 96050a2
1 parent bace279
commit 96050a2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/tensorflow_serving/servables/tensorflow/oss/resource_estimator.cc b/tensorflow_serving/servables/tensorflow/oss/resource_estimator.cc
@@ -22,15 +22,15 @@ limitations under the License.
 namespace tensorflow {
 namespace serving {
 
-Status EstimateMainRamBytesFromValidationResult(const string& path,
-                                                ResourceAllocation* estimate) {
+absl::Status EstimateMainRamBytesFromValidationResult(
+    const string& path, ResourceAllocation* estimate) {
   return errors::Unimplemented("Reading validation results is not supported");
 }
 
-Status EstimateMainRamBytesFromPath(const string& path,
-                                    bool use_validation_result,
-                                    FileProbingEnv* env,
-                                    ResourceAllocation* estimate) {
+absl::Status EstimateMainRamBytesFromPath(const string& path,
+                                          bool use_validation_result,
+                                          FileProbingEnv* env,
+                                          ResourceAllocation* estimate) {
   return EstimateResourceFromPathUsingDiskState(path, env, estimate);
 }
 

diff --git a/tensorflow_serving/servables/tensorflow/oss/resource_estimator_test.cc b/tensorflow_serving/servables/tensorflow/oss/resource_estimator_test.cc
@@ -46,15 +46,15 @@ class ResourceEstimatorTest : public ::testing::Test {
     // Set up the expectation that the directory contains exactly one child with
     // the given file size.
     EXPECT_CALL(env_, FileExists(export_dir_))
-        .WillRepeatedly(Return(Status()));
+        .WillRepeatedly(Return(absl::Status()));
     EXPECT_CALL(env_, GetChildren(export_dir_, _))
         .WillRepeatedly(DoAll(SetArgPointee<1>(std::vector<string>({child})),
-                              Return(Status())));
+                              Return(absl::Status())));
     EXPECT_CALL(env_, IsDirectory(child_path))
         .WillRepeatedly(Return(errors::FailedPrecondition("")));
     EXPECT_CALL(env_, GetFileSize(child_path, _))
         .WillRepeatedly(
-            DoAll(SetArgPointee<1>(file_size_), Return(Status())));
+            DoAll(SetArgPointee<1>(file_size_), Return(absl::Status())));
   }
 
   string export_dir_;