[CI][RFC] Replace black-jupyter with ruff-format (#37210)
Co-authored-by: Alison Shao <a.shao@wustl.edu>
This commit is contained in:
co-authored by
Alison Shao
parent
2641e427be
commit
28262c20df
@@ -309,7 +309,7 @@ class TestKimiK2DetectorSpecialTokenLeakage(unittest.TestCase):
|
||||
def test_no_leak_on_error_fallback(self):
|
||||
"""On parse errors, normal_text fallback has tokens stripped."""
|
||||
cleaned = _strip_special_tokens(
|
||||
"leaked<|tool_calls_section_begin|>" "<|tool_call_end|>content"
|
||||
"leaked<|tool_calls_section_begin|><|tool_call_end|>content"
|
||||
)
|
||||
self.assertEqual(cleaned, "leakedcontent")
|
||||
|
||||
@@ -992,7 +992,7 @@ class TestKimiK2EndToEnd(unittest.TestCase):
|
||||
"<|tool_call_begin|>functions.get_weather:1"
|
||||
f'<|tool_call_argument_begin|>{{"city":',
|
||||
' "Bad", "valid": fasle',
|
||||
"<|tool_call_end|>" "<|tool_calls_section_end|>" + good_section_1,
|
||||
"<|tool_call_end|><|tool_calls_section_end|>" + good_section_1,
|
||||
],
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user