From 1a7cdba4ec99001f1f013b44d51c1e2e96049e1b Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Wed, 23 Oct 2024 14:55:46 +0000 Subject: [PATCH] Update tests --- tests/test_quantization/lifecycle/test_apply.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_quantization/lifecycle/test_apply.py b/tests/test_quantization/lifecycle/test_apply.py index c0cacdf7..045eba94 100644 --- a/tests/test_quantization/lifecycle/test_apply.py +++ b/tests/test_quantization/lifecycle/test_apply.py @@ -27,7 +27,7 @@ from compressed_tensors.quantization.lifecycle import ( apply_quantization_config, apply_quantization_status, - find_compression_targets, + expand_targets, ) from compressed_tensors.quantization.utils import iter_named_leaf_modules from transformers import AutoModelForCausalLM @@ -296,6 +296,6 @@ def test_apply_quantization_status(caplog, ignore, should_raise_warning): ), ], ) -def test_find_compression_targets(model, targets, ignore, expected): - actual_targets = find_compression_targets(model, targets, ignore) +def test_expand_targets(model, targets, ignore, expected): + actual_targets = expand_targets(model, targets, ignore) assert actual_targets == expected