From 4c57287be7f98e269cb3ee6472b01844a57c6afd Mon Sep 17 00:00:00 2001 From: Grain Team Date: Wed, 25 Mar 2026 17:16:40 -0700 Subject: [PATCH] Internal PiperOrigin-RevId: 889512007 --- .../_src/python/dataset/transformations/BUILD | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/grain/_src/python/dataset/transformations/BUILD b/grain/_src/python/dataset/transformations/BUILD index aa832cc05..f3343c953 100644 --- a/grain/_src/python/dataset/transformations/BUILD +++ b/grain/_src/python/dataset/transformations/BUILD @@ -392,3 +392,29 @@ py_test( "@pypi//numpy:pkg", ], ) + +py_strict_test( + name = "imagenet_tmp_test", + srcs = ["imagenet_tmp_test.py"], + deps = [ + "//grain", + "//grain:tensorflow", + "//grain/_src/python:options", + "//grain/_src/python/dataset", + "//grain/_src/python/experimental/autotune/python:bindings", + "@abseil-py//absl/testing:absltest", + "@pypi//tensorflow:pkg", + ], +) + +py_strict_test( + name = "usl_sandbox_test", + srcs = ["usl_sandbox_test.py"], + deps = [ + "//grain", + "//grain/_src/python:options", + "//grain/_src/python/dataset", + "//grain/_src/python/experimental/autotune/python:bindings", + "@abseil-py//absl/testing:absltest", + ], +)