From 47338376a5fb63066a1ad25982d0a2744ef94765 Mon Sep 17 00:00:00 2001
From: zhulin1 <zhulin1@pjlab.org.cn>
Date: Mon, 1 Jul 2024 19:51:40 +0800
Subject: [PATCH] update

---
 tests/test_hf_model.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py
index 5429d3b..092cb59 100644
--- a/tests/test_hf_model.py
+++ b/tests/test_hf_model.py
@@ -233,7 +233,8 @@ class TestReward:
         rank_res = model.rank(tokenizer, [chat_1, chat_2])
         print('rank_res: ', rank_res)  # lower index means higher score
         # >>> rank_res:  [0, 1]
-        assert rank_res[0] == 0 & rank_res[1] == 1
+        assert rank_res[0] == 0
+        assert rank_res[1] == 1
 
     @pytest.mark.parametrize(
         'model_name',