Automated Code Change

PiperOrigin-RevId: 796979132

Author: tfx-copybara

struct2tensor/kernels/parquet/BUILD

@@ -60,6 +60,7 @@ cc_library(
     deps = [
         ":parent_indices_builder_lib",
         ":parquet_reader_util_lib",
+        "@com_google_absl//absl/status",
         "@com_google_absl//absl/strings",
         "@arrow//:arrow",
         "//struct2tensor/kernels:vector_to_tensor",

struct2tensor/kernels/parquet/parquet_reader.cc

@@ -16,6 +16,7 @@ limitations under the License.
 
 #include <memory>
 
+#include "absl/status/status.h"
 #include "absl/strings/str_cat.h"
 #include "struct2tensor/kernels/parquet/parquet_reader_util.h"
 #include "struct2tensor/kernels/vector_to_tensor.h"
@@ -440,7 +441,7 @@ tensorflow::Status ParquetReader::ReadOneColumnTemplated(
         column_index, def_levels, rep_levels, &cumulative_values);
 
     ++(*messages_read);
-    if (tensorflow::errors::IsOutOfRange(s)) {
+    if (absl::IsOutOfRange(s)) {
       break;
     }
     TF_RETURN_IF_ERROR(s);