From 2eb07972646aa4c303a4dee42e1d36f51df43bc4 Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Fri, 20 Mar 2026 21:43:39 -0700
Subject: [PATCH] Internal

PiperOrigin-RevId: 887129948
---
 grain/proto/BUILD | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/grain/proto/BUILD b/grain/proto/BUILD
index 6144d9441..23b4124b5 100644
--- a/grain/proto/BUILD
+++ b/grain/proto/BUILD
@@ -7,5 +7,8 @@ package(default_visibility = default_visibility)
 py_proto_library(
     name = "execution_summary_py_pb2",
     # For profiling tooling.
+    visibility = default_visibility + [
+        "//learning/gemini/cms/core/data/iterators:__subpackages__",
+        "//learning/gemini/gemax/core/data/iterators:__subpackages__",
     srcs = ["execution_summary.proto"],
 )