internal change

PiperOrigin-RevId: 702150045
google · Dec 3, 2024 · 6640613 · 6640613
1 parent 567b3d6
commit 6640613
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 12 deletions.
diff --git a/third_party/nucleus/io/gfile.cc b/third_party/nucleus/io/gfile.cc
@@ -41,13 +41,13 @@ namespace nucleus {
 
 bool Exists(const std::string& filename) {
   // FileExists sets s to tensorflow::error::NOT_FOUND if it doesn't exist.
-  absl::Status s = tensorflow::Env::Default()->FileExists(filename);
+  tensorflow::Status s = tensorflow::Env::Default()->FileExists(filename);
   return s.ok();
 }
 
 std::vector<std::string> Glob(const std::string& pattern) {
   std::vector<std::string> results;
-  absl::Status s =
+  ::tensorflow::Status s =
       tensorflow::Env::Default()->GetMatchingPaths(pattern, &results);
   return results;
 }
@@ -56,7 +56,7 @@ ReadableFile::ReadableFile() {}
 
 std::unique_ptr<ReadableFile> ReadableFile::New(const std::string& filename) {
   std::unique_ptr<tensorflow::RandomAccessFile> file;
-  absl::Status status =
+  tensorflow::Status status =
       tensorflow::Env::Default()->NewRandomAccessFile(filename, &file);
   if (!status.ok()) {
     return nullptr;
@@ -91,7 +91,8 @@ WritableFile::WritableFile() {}
 std::unique_ptr<WritableFile> WritableFile::New(const std::string& filename) {
   std::unique_ptr<tensorflow::WritableFile> file;
 
-  absl::Status s = tensorflow::Env::Default()->NewWritableFile(filename, &file);
+  tensorflow::Status s =
+      tensorflow::Env::Default()->NewWritableFile(filename, &file);
 
   if (!s.ok()) {
     return nullptr;
@@ -104,7 +105,7 @@ std::unique_ptr<WritableFile> WritableFile::New(const std::string& filename) {
 }
 
 bool WritableFile::Write(const std::string& s) {
-  absl::Status status = file_->Append(s);
+  tensorflow::Status status = file_->Append(s);
   return status.ok();
 }
 

diff --git a/third_party/nucleus/io/tfrecord_reader.cc b/third_party/nucleus/io/tfrecord_reader.cc
@@ -45,7 +45,7 @@ TFRecordReader::TFRecordReader() {}
 std::unique_ptr<TFRecordReader> TFRecordReader::New(
     const std::string& filename, const std::string& compression_type) {
   std::unique_ptr<tensorflow::RandomAccessFile> file;
-  absl::Status s =
+  tensorflow::Status s =
       tensorflow::Env::Default()->NewRandomAccessFile(filename, &file);
   if (!s.ok()) {
     LOG(ERROR) << s;
@@ -74,7 +74,7 @@ bool TFRecordReader::GetNext() {
     return false;
   }
 
-  absl::Status s = reader_->ReadRecord(&offset_, &record_);
+  tensorflow::Status s = reader_->ReadRecord(&offset_, &record_);
 
   return s.ok();
 }

diff --git a/third_party/nucleus/io/tfrecord_writer.cc b/third_party/nucleus/io/tfrecord_writer.cc
@@ -45,7 +45,8 @@ TFRecordWriter::TFRecordWriter() {}
 std::unique_ptr<TFRecordWriter> TFRecordWriter::New(
     const std::string& filename, const std::string& compression_type) {
   std::unique_ptr<tensorflow::WritableFile> file;
-  absl::Status s = tensorflow::Env::Default()->NewWritableFile(filename, &file);
+  tensorflow::Status s =
+      tensorflow::Env::Default()->NewWritableFile(filename, &file);
   if (!s.ok()) {
     LOG(ERROR) << s;
     return nullptr;
@@ -68,29 +69,29 @@ bool TFRecordWriter::WriteRecord(const std::string& record) {
   if (writer_ == nullptr) {
     return false;
   }
-  absl::Status s = writer_->WriteRecord(record);
+  tensorflow::Status s = writer_->WriteRecord(record);
   return s.ok();
 }
 
 bool TFRecordWriter::Flush() {
   if (writer_ == nullptr) {
     return false;
   }
-  absl::Status s = writer_->Flush();
+  tensorflow:: Status s = writer_->Flush();
   return s.ok();
 }
 
 bool TFRecordWriter::Close() {
   if (writer_ != nullptr) {
-    absl::Status s = writer_->Close();
+    tensorflow::Status s = writer_->Close();
     if (!s.ok()) {
       return false;
     }
     writer_ = nullptr;
   }
 
   if (file_ != nullptr) {
-    absl::Status s = file_->Close();
+    tensorflow:: Status s = file_->Close();
     if (!s.ok()) {
       return false;
     }