From ffc9a86c2654df314fed3396b5760f1967ee9554 Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Wed, 17 Jul 2024 09:31:21 -0700
Subject: [PATCH] Internal change.

PiperOrigin-RevId: 653260476
---
 grain/python_proto.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/grain/python_proto.py b/grain/python_proto.py
index fe929386..434ebd6a 100644
--- a/grain/python_proto.py
+++ b/grain/python_proto.py
@@ -21,3 +21,5 @@
 
 from ._src.python.proto.decode import parse_tf_example
 from ._src.python.proto.encode import make_tf_example
+
+from ._src.python.proto import lazy_np