From 78e0e6b0585c85efdba7a386134cd324624c1f9c Mon Sep 17 00:00:00 2001
From: Yash Katariya <yashkatariya@google.com>
Date: Thu, 7 Dec 2023 11:30:08 -0800
Subject: [PATCH] Internal change

PiperOrigin-RevId: 588856820
---
 tests/BUILD | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/BUILD b/tests/BUILD
index 29776452d..dd28fae50 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -1365,6 +1365,7 @@ exports_files(
         "python_callback_test.py",
         "shard_map_test.py",
         "transfer_guard_test.py",
+        "layout_test.py",
     ],
     visibility = jax_test_file_visibility,
 )