From afa62b7c7901ffdb3af64305540d29a772782b76 Mon Sep 17 00:00:00 2001
From: Qubitium <qubitium@modelcloud.ai>
Date: Sat, 1 Nov 2025 13:52:39 +0000
Subject: [PATCH] fix ci test

---
 tests/models/model_test.py               | 4 ++--
 tests/models/test_act_group_aware.py     | 2 +-
 tests/models/test_ling.py                | 2 +-
 tests/models/test_multi_vs_single_gpu.py | 2 +-
 tests/{test_gptqv2.py => test_gptaq.py}  | 2 +-
 5 files changed, 6 insertions(+), 6 deletions(-)
 rename tests/{test_gptqv2.py => test_gptaq.py} (97%)

diff --git a/tests/models/model_test.py b/tests/models/model_test.py
index cae636155..cae7b2fce 100644
--- a/tests/models/model_test.py
+++ b/tests/models/model_test.py
@@ -107,7 +107,7 @@ class ModelTest(unittest.TestCase):
     GROUP_SIZE = 128
     DESC_ACT = False
     SYM = True
-    V2 = False
+    GPTQA = False
     ACT_GROUP_AWARE = True
     FAIL_SAFE = True
     EORA = None
@@ -783,7 +783,7 @@ def quantModel(self, model_id_or_path, trust_remote_code=False, dtype="auto", ne
             act_group_aware=self.ACT_GROUP_AWARE,
             fail_safe=self.FAIL_SAFE,
             sym=self.SYM,
-            v2=self.V2,
+            gptaq=self.GPTQA,
             adapter=self.EORA,
             pack_impl="cpu",
             vram_strategy=self.VRAM_STRATEGY,
diff --git a/tests/models/test_act_group_aware.py b/tests/models/test_act_group_aware.py
index dffcc77a0..5a58f2c29 100644
--- a/tests/models/test_act_group_aware.py
+++ b/tests/models/test_act_group_aware.py
@@ -17,7 +17,7 @@ class TestHybridActOrder(ModelTest):
             "acc_norm": {"value": 0.3439, "floor_pct": 0.05},
         },
     }
-    V2 = False
+    GPTQA = False
     ACT_GROUP_AWARE = True
 
     def test_llama3_2(self):
diff --git a/tests/models/test_ling.py b/tests/models/test_ling.py
index 5303f5e06..cfa1ba9a1 100644
--- a/tests/models/test_ling.py
+++ b/tests/models/test_ling.py
@@ -19,7 +19,7 @@ class TestLing(ModelTest):
     }
     TRUST_REMOTE_CODE = True
     # EVAL_BATCH_SIZE = 6
-    V2 = False
+    GPTQA = False
     DEBUG = True
     ACT_GROUP_AWARE = True
     DESC_ACT = False
diff --git a/tests/models/test_multi_vs_single_gpu.py b/tests/models/test_multi_vs_single_gpu.py
index 8394ea610..f53b15ae9 100644
--- a/tests/models/test_multi_vs_single_gpu.py
+++ b/tests/models/test_multi_vs_single_gpu.py
@@ -53,7 +53,7 @@ class TestMultiVsSingleGPU(ModelTest):
             "acc_norm": {"value": 0.3549, "floor_pct": 0.05},
         },
     }
-    V2 = False
+    GPTQA = False
     DEBUG = True
     ACT_GROUP_AWARE = False
     DESC_ACT = True
diff --git a/tests/test_gptqv2.py b/tests/test_gptaq.py
similarity index 97%
rename from tests/test_gptqv2.py
rename to tests/test_gptaq.py
index 7aed4695d..5adab89e0 100644
--- a/tests/test_gptqv2.py
+++ b/tests/test_gptaq.py
@@ -19,7 +19,7 @@ class TestQwen2_5_GPTQv2(ModelTest):
     }
     TRUST_REMOTE_CODE = False
     EVAL_BATCH_SIZE = 6
-    V2 = True
+    GPTQA = True
 
     def test_qwen2_5(self):
         self.quant_lm_eval()