update

2024-07-01 19:51:40 +08:00 · 2024-07-01 19:51:40 +08:00 · 47338376a5
parent d4d54639c6
commit 47338376a5
1 changed files with 2 additions and 1 deletions
--- 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',