From cdb0f34f10f5407e83766a8c5c9c5aff6af2a84e Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Fri, 19 Jul 2024 01:17:37 +0800
Subject: [PATCH] fix test

---
 tests/data/processors/test_unsupervised.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/tests/data/processors/test_unsupervised.py b/tests/data/processors/test_unsupervised.py
index 8713c772..1bfab53e 100644
--- a/tests/data/processors/test_unsupervised.py
+++ b/tests/data/processors/test_unsupervised.py
@@ -30,17 +30,18 @@ TINY_DATA = os.environ.get("TINY_DATA", "llamafactory/tiny-supervised-dataset")
 
 TRAIN_ARGS = {
     "model_name_or_path": TINY_LLAMA,
-    "stage": "sft",
-    "do_predict": True,
+    "stage": "ppo",
+    "do_train": True,
     "finetuning_type": "full",
-    "eval_dataset": "system_chat",
+    "reward_model": "",
+    "reward_model_type": "full",
+    "dataset": "system_chat",
     "dataset_dir": "REMOTE:" + DEMO_DATA,
     "template": "llama3",
     "cutoff_len": 8192,
     "overwrite_cache": True,
     "output_dir": "dummy_dir",
     "overwrite_output_dir": True,
-    "predict_with_generate": True,
     "fp16": True,
 }