Skip to content

Commit 96050a2

Browse files
Automated Code Change
PiperOrigin-RevId: 690971099
1 parent bace279 commit 96050a2

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

tensorflow_serving/servables/tensorflow/oss/resource_estimator.cc

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -22,15 +22,15 @@ limitations under the License.
2222
namespace tensorflow {
2323
namespace serving {
2424

25-
Status EstimateMainRamBytesFromValidationResult(const string& path,
26-
ResourceAllocation* estimate) {
25+
absl::Status EstimateMainRamBytesFromValidationResult(
26+
const string& path, ResourceAllocation* estimate) {
2727
return errors::Unimplemented("Reading validation results is not supported");
2828
}
2929

30-
Status EstimateMainRamBytesFromPath(const string& path,
31-
bool use_validation_result,
32-
FileProbingEnv* env,
33-
ResourceAllocation* estimate) {
30+
absl::Status EstimateMainRamBytesFromPath(const string& path,
31+
bool use_validation_result,
32+
FileProbingEnv* env,
33+
ResourceAllocation* estimate) {
3434
return EstimateResourceFromPathUsingDiskState(path, env, estimate);
3535
}
3636

tensorflow_serving/servables/tensorflow/oss/resource_estimator_test.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,15 +46,15 @@ class ResourceEstimatorTest : public ::testing::Test {
4646
// Set up the expectation that the directory contains exactly one child with
4747
// the given file size.
4848
EXPECT_CALL(env_, FileExists(export_dir_))
49-
.WillRepeatedly(Return(Status()));
49+
.WillRepeatedly(Return(absl::Status()));
5050
EXPECT_CALL(env_, GetChildren(export_dir_, _))
5151
.WillRepeatedly(DoAll(SetArgPointee<1>(std::vector<string>({child})),
52-
Return(Status())));
52+
Return(absl::Status())));
5353
EXPECT_CALL(env_, IsDirectory(child_path))
5454
.WillRepeatedly(Return(errors::FailedPrecondition("")));
5555
EXPECT_CALL(env_, GetFileSize(child_path, _))
5656
.WillRepeatedly(
57-
DoAll(SetArgPointee<1>(file_size_), Return(Status())));
57+
DoAll(SetArgPointee<1>(file_size_), Return(absl::Status())));
5858
}
5959

6060
string export_dir_;

0 commit comments

Comments
 (0)