fix TestApplyQuantization test

Signed-off-by: Kyle Sayers <[email protected]>

Author: kylesayrs

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

@@ -5,7 +5,6 @@
 from parameterized import parameterized
 
 from llmcompressor.modifiers.obcq import SparseGPTModifier
-from llmcompressor.modifiers.quantization.calibration import calibrate_input_hook
 from llmcompressor.modifiers.quantization.gptq import GPTQModifier
 from tests.llmcompressor.modifiers.conf import (
     LifecyleTestingHarness,
@@ -61,7 +60,7 @@ def test_successful_layerwise_recipe(self):
 
 
 @pytest.mark.unit
-class TestCreateDefaultQuantModifier(unittest.TestCase):
+class TestApplyQuantization(unittest.TestCase):
     def setUp(self):
         setup_modifier_factory()
 
@@ -77,8 +76,10 @@ def test_create_default_quant_modifier(self):
                 assert hasattr(module, "quantization_scheme")
                 assert hasattr(module, "input_observer")
                 assert hasattr(module, "weight_observer")
-                assert module._forward_pre_hooks[0] is calibrate_input_hook
-                assert module._forward_hooks[0] is modifier.calibrate_module
+                pre_hooks = list(module._forward_pre_hooks.values())
+                post_hooks = list(module._forward_hooks.values())
+                assert pre_hooks[0].__name__ == "calibrate_input_hook"
+                assert post_hooks[0].__name__ == "calibrate_module"
 
 
 class TestSetQuantInGPTQ(unittest.TestCase):