Skip to content

Commit 21ee885

Browse files
Copilotjustinchuby
andcommitted
Remove * import from evaluator.py and add explicit exports
Co-authored-by: justinchuby <[email protected]>
1 parent 03b60ec commit 21ee885

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

onnxscript/evaluator.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,20 @@
55
This module re-exports the public API from the internal evaluator module.
66
"""
77

8-
# Re-export all symbols from the internal evaluator module
9-
from onnxscript._internal.evaluator import * # noqa: F403
10-
11-
# Explicitly list main exports for type checking
128
from onnxscript._internal.evaluator import (
139
Evaluator,
10+
OnnxReferenceRuntimeEvaluator,
11+
ORTEvaluator,
12+
ORTMixedEvaluator,
1413
default,
1514
default_as,
1615
)
1716

18-
__all__ = ["default", "default_as", "Evaluator"]
17+
__all__ = [
18+
"Evaluator",
19+
"ORTEvaluator",
20+
"ORTMixedEvaluator",
21+
"OnnxReferenceRuntimeEvaluator",
22+
"default",
23+
"default_as",
24+
]

0 commit comments

Comments
 (0)