Skip to content

Commit 60a536a

Browse files
committed
update tests
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 4b5fe76 commit 60a536a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/llmcompressor/pytorch/modifiers/pruning/constant/test_pytorch.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -74,15 +74,14 @@ def test_constant_pruning_modifier_e2e(model, optimizer):
7474
state.update(
7575
model=model,
7676
optimizer=optimizer(model.parameters(), lr=0.1),
77-
start=0,
7877
)
7978
modifier = ConstantPruningModifier(
8079
targets="__ALL_PRUNABLE__",
8180
start=0,
8281
end=1,
8382
update=0.5,
8483
)
85-
modifier.initialize(state)
84+
modifier.initialize(state, start=0)
8685

8786
# check mask is added and has correct sparsity
8887

0 commit comments

Comments
 (0)