From b91949b918748cf068ffb4124707320a93c8f14e Mon Sep 17 00:00:00 2001
From: zhulinJulia24 <145004780+zhulinJulia24@users.noreply.github.com>
Date: Tue, 11 Jun 2024 19:29:26 +0800
Subject: [PATCH] Add newest math model into daily testcase (#744)

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

diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py
index b6118a5..f6e5b45 100644
--- a/tests/test_hf_model.py
+++ b/tests/test_hf_model.py
@@ -127,7 +127,9 @@ class TestMath:
 
     @pytest.mark.parametrize(
         'model_name',
-        ['internlm/internlm2-math-7b', 'internlm/internlm2-math-base-7b'],
+        ['internlm/internlm2-math-7b', 'internlm/internlm2-math-base-7b',
+         'internlm/internlm2-math-plus-1_8b', 'internlm/internlm2-math-plus-7b'
+        ],
     )
     @pytest.mark.parametrize(
         'usefast',