From e790e79c58114ffa318400ea21ccdb7caa1b97fe Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Wed, 11 Mar 2026 10:19:49 -0700
Subject: [PATCH] Internal

PiperOrigin-RevId: 882080624
---
 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"],
 )