Skip to content

Commit a105dee

Browse files
Add missing __init__.py (#2115)
SUMMARY: "Address #2111. Checked other directories and that was the only missing `__init__.py`" TEST PLAN: "`pip install --no-build-isolation .` and checked `from llmcompressor.pipelines.sequential.ast_utils.auto_wrapper import AutoWrapper`" --------- Signed-off-by: Jaemin Son <woalsdnd@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 6d27dc3 commit a105dee

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

src/llmcompressor/pipelines/sequential/ast_helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
import torch
1111
from compressed_tensors.utils import patch_attr
1212

13-
from llmcompressor.pipelines.sequential.ast_utils.auto_wrapper import AutoWrapper
13+
from llmcompressor.pipelines.sequential.ast_utils import AutoWrapper
1414

1515
__all__ = ["autowrap_forwards", "append_autowrap_source_on_fail"]
1616

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from .auto_wrapper import AutoWrapper
2+
from .control_flow_analyzer import ControlFlowAnalyzer
3+
from .name_analyzer import NameAnalyzer
4+
5+
__all__ = [
6+
"AutoWrapper",
7+
"ControlFlowAnalyzer",
8+
"NameAnalyzer",
9+
]

0 commit comments

Comments
 (0)