diff --git a/tensorflow_serving/servables/tensorflow/oss/resource_estimator.cc b/tensorflow_serving/servables/tensorflow/oss/resource_estimator.cc index e0fdab0c22a..8d8d3d35034 100644 --- 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 index 7d42b2ce33a..c29160875ca 100644 --- 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({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_;