Skip to content

Commit 129c793

Browse files
authored
[Tests] Fix tests (#2140)
## Purpose ## * Fix tests ## Changes ## * Remove unnecessary check for initialization on spinquant and quip modifiers --------- Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent a5f7031 commit 129c793

File tree

3 files changed

+2
-7
lines changed

3 files changed

+2
-7
lines changed

src/llmcompressor/modifiers/transform/quip/base.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -132,9 +132,6 @@ def on_finalize(self, state: State, **kwargs) -> bool:
132132
return True
133133

134134
def _get_targets(self, model: torch.nn.Module) -> NamedModules:
135-
if not self.initialized_:
136-
raise ValueError("Cannot get targets before modifier has been initialized")
137-
138135
return [
139136
(name, module)
140137
for scheme in self.transform_config.config_groups.values()

src/llmcompressor/modifiers/transform/spinquant/base.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -185,9 +185,6 @@ def on_finalize(self, state: State, **kwargs) -> bool:
185185
return True
186186

187187
def _get_targets(self, model: torch.nn.Module) -> NamedModules:
188-
if not self.initialized_:
189-
raise ValueError("Cannot get targets before modifier has been initialized")
190-
191188
return [
192189
(name, module)
193190
for scheme in self.transform_config.config_groups.values()

tests/llmcompressor/modifiers/transform/test_correctness.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,8 @@ def test_apply_correctness(
4545
with torch.no_grad():
4646
true_output = model(**input)
4747

48-
modifier.initialize(state)
48+
modifier.on_initialize(state)
49+
modifier.on_start(state, None)
4950

5051
with torch.no_grad():
5152
output = model(**input)

0 commit comments

Comments
 (0)