Skip to content

Commit 6bfbb4d

Browse files
gramalingamCopilot
andauthored
Expose OpBuilder as onnxscript.OpBuilder (#2840)
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>
1 parent 1350ed2 commit 6bfbb4d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

onnxscript/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
"OnnxFunction",
1414
"TracedOnnxFunction",
1515
"GraphBuilder",
16+
"OpBuilder",
1617
"proto2python",
1718
"external_tensor",
1819
"BFLOAT16",
@@ -129,7 +130,7 @@
129130
# isort: on
130131

131132
from . import ir, nn, optimizer, rewriter, version_converter
132-
from ._internal.builder import GraphBuilder
133+
from ._internal.builder import GraphBuilder, OpBuilder
133134
from ._internal.utils import external_tensor
134135
from ._internal.values import OnnxFunction, TracedOnnxFunction
135136

0 commit comments

Comments
 (0)