From b71a23e580ca3b15d3109d1474815693334efc5d Mon Sep 17 00:00:00 2001 From: G Ramalingam Date: Fri, 8 May 2026 17:18:50 +0000 Subject: [PATCH] Expose build_function and build_graph in public API Add build_function and build_graph to onnxscript's top-level exports, allowing downstream packages (e.g. mobius) to import them directly as onnxscript.build_function / onnxscript.build_graph instead of reaching into onnxscript._internal.builder. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> Signed-off-by: G Ramalingam --- onnxscript/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/onnxscript/__init__.py b/onnxscript/__init__.py index 5b745807d3..829ee212cf 100644 --- a/onnxscript/__init__.py +++ b/onnxscript/__init__.py @@ -16,6 +16,8 @@ "OpBuilder", "OpBuilderBase", "TapeBuilder", + "build_function", + "build_graph", "proto2python", "external_tensor", "BFLOAT16", @@ -132,7 +134,7 @@ # isort: on from . import ir, nn, optimizer, rewriter, version_converter -from ._internal.builder import GraphBuilder, OpBuilder +from ._internal.builder import GraphBuilder, OpBuilder, build_function, build_graph from ._internal.tape_builder import OpBuilderBase, TapeBuilder from ._internal.utils import external_tensor from ._internal.values import OnnxFunction, TracedOnnxFunction