fix a previous issue

cleanlab · Aug 31, 2024 · e255361 · e255361
1 parent 861d713
commit e255361
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tlm/test_validation.py b/tests/tlm/test_validation.py
@@ -184,7 +184,7 @@ def test_prompt_too_long_exception_batch_score(tlm: TLM, num_prompts: int):
         "a" * (MAX_PROMPT_LENGTH_TOKENS // 2 + 1) * CHARACTERS_PER_TOKEN
     )
     responses[combined_too_long_index] = (
-        "a" * (MAX_PROMPT_LENGTH_TOKENS // 2 + 1) * CHARACTERS_PER_TOKEN
+        "a" * (MAX_RESPONSE_LENGTH_TOKENS // 2 + 1) * CHARACTERS_PER_TOKEN
     )
 
     with pytest.raises(