From 87c9dd55410d6970794c99964b845843075ce202 Mon Sep 17 00:00:00 2001 From: Orbax Authors Date: Mon, 15 Sep 2025 08:50:24 -0700 Subject: [PATCH] Internal adjustment to tree-verity reporting PiperOrigin-RevId: 807257406 --- export/orbax/export/tensorflow_export.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/export/orbax/export/tensorflow_export.py b/export/orbax/export/tensorflow_export.py index 8470beec4..0feb7ab3d 100644 --- a/export/orbax/export/tensorflow_export.py +++ b/export/orbax/export/tensorflow_export.py @@ -26,6 +26,8 @@ from orbax.export import utils import tensorflow as tf +from .security.bcid.tree_verity.public import tree_verity_pb2 + obx_export_config = config.config maybe_reraise = reraise_utils.maybe_reraise