From 563ccd3dafe43040907467d6a21c73eea6626a3b Mon Sep 17 00:00:00 2001
From: ArrayRecord Team <array-record@google.com>
Date: Fri, 19 Jun 2026 10:55:30 -0700
Subject: [PATCH] Internal

PiperOrigin-RevId: 934987026
---
 python/BUILD          | 26 ++++++++++++++++++++++++++
 python/testdata/BUILD |  5 ++++-
 2 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/python/BUILD b/python/BUILD
index d247e61..2469daf 100644
--- a/python/BUILD
+++ b/python/BUILD
@@ -2,6 +2,7 @@
 
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
 load("@pypi//:requirements.bzl", "requirement")
+load("//third_party/bazel_rules/rules_python/python:py_binary.bzl", "py_binary")
 
 package(default_visibility = ["//visibility:public"])
 
@@ -34,6 +35,20 @@ py_test(
     ],
 )
 
+py_test(
+    name = "read_my_data_test",
+    srcs = ["read_my_data_test.py"],
+    data = [
+        ":array_record_module.so",
+        "//python/testdata:fine_tuning_train_multilabel.array_record",
+    ],
+    deps = [
+        "//third_party/py/requests",
+        "//third_party/py/tensorflow",
+        "@abseil-py//absl/testing:absltest",
+    ],
+)
+
 py_library(
     name = "array_record_data_source",
     srcs = ["array_record_data_source.py"],
@@ -59,3 +74,14 @@ py_test(
         "@abseil-py//absl/testing:parameterized",
     ],
 )
+
+py_binary(
+    name = "read_my_data_bin",
+    srcs = ["read_my_data_test.py"],
+    data = [":array_record_module.so"],
+    main = "read_my_data_test.py",
+    deps = [
+        "//third_party/py/tensorflow",
+        "@abseil-py//absl/testing:absltest",
+    ],
+)
diff --git a/python/testdata/BUILD b/python/testdata/BUILD
index 54bb15c..2fbca92 100644
--- a/python/testdata/BUILD
+++ b/python/testdata/BUILD
@@ -2,4 +2,7 @@ package(default_visibility = ["//visibility:public"])
 
 licenses(["notice"])
 
-exports_files(glob(["*.array_record-*"]))
+exports_files(glob([
+    "*.array_record-*",
+    "fine_tuning_train_multilabel.array_record",
+]))