Update tests

Signed-off-by: DarkLight1337 <[email protected]>

tests/models/embedding/language/test_cls_models.py

@@ -1,7 +1,4 @@
-"""Compare the outputs of HF and vLLM when using greedy sampling.
-
-This test only tests small models. Big models such as 7B should be tested from
-test_big_models.py because it could use a larger instance to run tests.
+"""Compare the classification outputs of HF and vLLM models.
 
 Run `pytest tests/models/test_cls_models.py`.
 """
@@ -13,6 +10,8 @@
 @pytest.mark.parametrize(
     "model",
     [
+        pytest.param("ai21labs/Jamba-tiny-reward-dev",
+                     marks=[pytest.mark.core_model]),
         pytest.param("jason9693/Qwen2.5-1.5B-apeach",
                      marks=[pytest.mark.core_model, pytest.mark.cpu_model]),
     ],

tests/models/embedding/language/test_scoring.py

@@ -1,6 +1,6 @@
-"""Compare the embedding outputs of HF and vLLM models.
+"""Compare the scoring outputs of HF and vLLM models.
 
-Run `pytest tests/models/embedding/language/test_embedding.py`.
+Run `pytest tests/models/embedding/language/test_scoring.py`.
 """
 import math