From 2e845fde3d77333b3f295a296f87681bde644057 Mon Sep 17 00:00:00 2001 From: Ganesan Ramalingam Date: Wed, 4 Mar 2026 09:55:53 -0800 Subject: [PATCH] Expose OpBuilder as onnxscript.OpBuilder Add OpBuilder to __all__ and import it from _internal.builder in the top-level onnxscript package. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> --- onnxscript/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/onnxscript/__init__.py b/onnxscript/__init__.py index 9783e37c24..6731631aae 100644 --- a/onnxscript/__init__.py +++ b/onnxscript/__init__.py @@ -13,6 +13,7 @@ "OnnxFunction", "TracedOnnxFunction", "GraphBuilder", + "OpBuilder", "proto2python", "external_tensor", "BFLOAT16", @@ -129,7 +130,7 @@ # isort: on from . import ir, nn, optimizer, rewriter, version_converter -from ._internal.builder import GraphBuilder +from ._internal.builder import GraphBuilder, OpBuilder from ._internal.utils import external_tensor from ._internal.values import OnnxFunction, TracedOnnxFunction