Skip to content

Commit e211825

Browse files
Copilotjustinchuby
andauthored
Expose DeduplicateInitializersPass in common passes module (#99)
This PR exposes the `DeduplicateInitializersPass` in the `onnx_ir.passes.common` module by adding it to the `__all__` list and importing it in the module's `__init__.py`. ## Changes Made - Added `"DeduplicateInitializersPass"` to the `__all__` list in alphabetical order - Added import statement: `from onnx_ir.passes.common.initializer_deduplication import (DeduplicateInitializersPass,)` ## Before ```python # This would fail with ImportError from onnx_ir.passes.common import DeduplicateInitializersPass # Users had to import directly from the submodule from onnx_ir.passes.common.initializer_deduplication import DeduplicateInitializersPass ``` ## After ```python # Now works as expected from onnx_ir.passes.common import DeduplicateInitializersPass # Pass can be used immediately pass_instance = DeduplicateInitializersPass() result = pass_instance(model) ``` The change follows the existing pattern used by other passes in the module and maintains alphabetical ordering in both the `__all__` list and import statements. Fixes #98. <!-- START COPILOT CODING AGENT TIPS --> --- 💬 Share your feedback on Copilot coding agent for the chance to win a $200 gift card! Click [here](https://survey.alchemer.com/s3/8343779/Copilot-Coding-agent) to start the survey. --------- Co-authored-by: copilot-swe-agent[bot] <[email protected]> Co-authored-by: justinchuby <[email protected]>
1 parent 62e750c commit e211825

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/onnx_ir/passes/common/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
"CheckerPass",
77
"ClearMetadataAndDocStringPass",
88
"CommonSubexpressionEliminationPass",
9+
"DeduplicateInitializersPass",
910
"InlinePass",
1011
"LiftConstantsToInitializersPass",
1112
"LiftSubgraphInitializersToMainGraphPass",
@@ -29,6 +30,9 @@
2930
LiftSubgraphInitializersToMainGraphPass,
3031
RemoveInitializersFromInputsPass,
3132
)
33+
from onnx_ir.passes.common.initializer_deduplication import (
34+
DeduplicateInitializersPass,
35+
)
3236
from onnx_ir.passes.common.inliner import InlinePass
3337
from onnx_ir.passes.common.onnx_checker import CheckerPass
3438
from onnx_ir.passes.common.shape_inference import ShapeInferencePass

0 commit comments

Comments
 (0)