diff --git a/tests/kernels/test_awq.py b/tests/kernels/test_awq.py index 212e0e20a..46fd4d1ed 100644 --- a/tests/kernels/test_awq.py +++ b/tests/kernels/test_awq.py @@ -341,6 +341,7 @@ def _build_torch_awq_module( module.post_init() return module + @classmethod def _build_exllama_v2_module( cls, qweight_cpu: torch.Tensor,