From 2e73fdadbb217629740e6d73169bb253b3c65478 Mon Sep 17 00:00:00 2001
From: Qubitium <qubitium@modelcloud.ai>
Date: Sat, 14 Mar 2026 00:54:37 +0000
Subject: [PATCH] missing arg

---
 tests/kernels/test_awq.py | 1 +
 1 file changed, 1 insertion(+)

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,