Automated Code Change

PiperOrigin-RevId: 627300223

Author: tfx-copybara

struct2tensor/kernels/parquet/parent_indices_builder.cc

@@ -38,7 +38,7 @@ tensorflow::Status ParentIndicesBuilder::Create(
 
   *parent_indices_builder =
       absl::WrapUnique(new ParentIndicesBuilder(repetition_pattern));
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 const std::vector<std::vector<tensorflow::int64>>&

struct2tensor/kernels/parquet/parquet_dataset_kernel.cc

@@ -70,7 +70,7 @@ class Dataset : public tensorflow::data::DatasetBase {
 
   tensorflow::Status CheckExternalState() const
   {
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
  protected:
@@ -108,7 +108,7 @@ class Dataset : public tensorflow::data::DatasetBase {
       tensorflow::mutex_lock l(mu_);
       if (current_file_index_ >= filenames_.size()) {
         *end_of_sequence = true;
-        return tensorflow::OkStatus();
+        return absl::OkStatus();
       }
 
       if (!parquet_reader_) {
@@ -167,7 +167,7 @@ class Dataset : public tensorflow::data::DatasetBase {
             std::move(parent_indices_and_values[column_index].values));
       }
 
-      return tensorflow::OkStatus();
+      return absl::OkStatus();
     }
 
    protected:

struct2tensor/kernels/parquet/parquet_reader.cc

@@ -52,7 +52,7 @@ class PeekableColumnReader : public PeekableColumnReaderBase {
     *pcr = absl::WrapUnique(
         new PeekableColumnReader<ParquetDataType>(column_index, file_reader));
     TF_RETURN_IF_ERROR(pcr->get()->Advance());
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
   PeekableColumnReader<ParquetDataType>& operator=(
@@ -113,7 +113,7 @@ class PeekableColumnReader : public PeekableColumnReaderBase {
           "Expected to read 1 level. Actually read %d level", levels_read);
     }
     value_exists_ = (values_read == 1);
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
 
  private:
@@ -169,7 +169,7 @@ tensorflow::Status GetColumnIndex(const std::string& column_name,
         absl::StrCat("Column not found: ", column_name));
   }
 
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 // Creates the repetition pattern for a path. For example "Document.DocID"
@@ -217,7 +217,7 @@ tensorflow::Status PopulatePeekableColumnReadersVector(
   TF_RETURN_IF_ERROR(internal::PeekableColumnReader<ParquetDataType>::Create(
       column_index, file_reader, &pcr));
   peekable_column_readers->push_back(std::move(pcr));
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 tensorflow::Status ParquetReader::Create(
@@ -301,7 +301,7 @@ tensorflow::Status ParquetReader::Create(
       value_paths, value_dtypes, batch_size, column_indices,
       std::move(file_reader), std::move(peekable_column_readers),
       std::move(parent_indices_builders)));
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 ParquetReader::ParquetReader(
@@ -354,9 +354,9 @@ tensorflow::Status ParquetReader::ReadMessages(
   total_rows_read_ += prev_column_messages_read;
   if (total_rows_read_ >= file_reader_->metadata()->num_rows()) {
     *end_of_file = true;
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   }
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 tensorflow::Status ParquetReader::ReadOneColumn(
@@ -425,7 +425,7 @@ tensorflow::Status ParquetReader::ReadOneColumn(
   parent_indices_and_values->emplace_back(
       ParentIndicesAndValues{parent_indices, value_tensor[0]});
 
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 template <typename ParquetDataType, typename T>
@@ -450,7 +450,7 @@ tensorflow::Status ParquetReader::ReadOneColumnTemplated(
   struct2tensor::VectorToTensor(cumulative_values, &res,
                                 /*produce_string_view=*/false);
   value_tensor->push_back(res);
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 template <typename ParquetDataType, typename T>
@@ -479,7 +479,7 @@ tensorflow::Status ParquetReader::ReadOneMessageFromOneColumn(
   } while (repetition_level > 0 &&
            repetition_level < max_repetition_level_[column_index]);
 
-  return tensorflow::OkStatus();
+  return absl::OkStatus();
 }
 
 }  // namespace parquet_dataset

struct2tensor/kernels/parquet/parquet_reader_util.cc

@@ -24,7 +24,7 @@ tensorflow::Status OpenFileWithStatus(
     std::unique_ptr<parquet::ParquetFileReader>* file_reader) {
   try {
     *file_reader = parquet::ParquetFileReader::OpenFile(filename, false);
-    return tensorflow::OkStatus();
+    return absl::OkStatus();
   } catch (const parquet::ParquetException& e) {
     return tensorflow::errors::Internal(
         absl::StrCat("Invalid File: ", filename));

struct2tensor/proto/BUILD

@@ -1,11 +1,11 @@
+load("//struct2tensor:struct2tensor.bzl", "s2t_proto_library", "s2t_proto_library_cc", "s2t_proto_library_py")
+
 package(
     default_visibility = ["@//:__subpackages__"],
 )
 
 licenses(["notice"])
 
-load("//struct2tensor:struct2tensor.bzl", "s2t_proto_library", "s2t_proto_library_cc", "s2t_proto_library_py")
-
 s2t_proto_library(
     name = "query_metadata_proto",
     srcs = ["query_metadata.proto"],