From 0e175f8ec91b1191b396bd6ed6bbe8687c0754a8 Mon Sep 17 00:00:00 2001
From: Kaihui-intel <kaihui.tang@intel.com>
Date: Mon, 10 Nov 2025 00:11:24 -0500
Subject: [PATCH] fix cuda CI

Signed-off-by: Kaihui-intel <kaihui.tang@intel.com>
---
 test/test_cuda/test_main_func.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/test/test_cuda/test_main_func.py b/test/test_cuda/test_main_func.py
index 523fe6784..571fc10f5 100644
--- a/test/test_cuda/test_main_func.py
+++ b/test/test_cuda/test_main_func.py
@@ -212,7 +212,6 @@ def test_low_cpu_mem_usage(self):
             group_size=group_size,
             iters=2,
             seqlen=10,
-            dataset=self.llm_dataloader,
             low_cpu_mem_usage=True,
         )
         autoround.quantize_and_save(output_dir=quantized_model_path, format="auto_round")