From 6d5c8db6b5960b65a29eef01ab0532764520a329 Mon Sep 17 00:00:00 2001
From: LRL2-ModelCloud <lrl2@modelcloud.ai>
Date: Wed, 15 Oct 2025 15:44:29 +0800
Subject: [PATCH] use GSM8K_COT

---
 tests/test_simple_quant.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_simple_quant.py b/tests/test_simple_quant.py
index 2d7871d1c..944b0ac15 100644
--- a/tests/test_simple_quant.py
+++ b/tests/test_simple_quant.py
@@ -81,7 +81,7 @@ def get_calib_data(tokenizer, rows: int):
 with tempfile.TemporaryDirectory() as tmp_dir:
     results = GPTQModel.eval(
         QUANT_SAVE_PATH,
-        tasks=[EVAL.LM_EVAL.GSM8K_PLATINUM_COT], #, EVAL.LM_EVAL.GSM8K_PLATINUM_COT],
+        tasks=[EVAL.LM_EVAL.GSM8K_COT], #, EVAL.LM_EVAL.GSM8K_PLATINUM_COT],
         apply_chat_template=True,
         random_seed=898,
         output_path= tmp_dir,