[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
@@ -101,7 +101,7 @@ class TestSkipTokenizerInit(CustomTestCase):
|
||||
self.assertEqual(
|
||||
len(item["meta_info"]["input_token_logprobs"]),
|
||||
num_input_logprobs,
|
||||
f'{len(item["meta_info"]["input_token_logprobs"])} mismatch with {len(input_ids)}',
|
||||
f"{len(item['meta_info']['input_token_logprobs'])} mismatch with {len(input_ids)}",
|
||||
)
|
||||
self.assertEqual(
|
||||
len(item["meta_info"]["output_token_logprobs"]),
|
||||
@@ -245,9 +245,9 @@ class TestSkipTokenizerInitVLM(TestSkipTokenizerInit):
|
||||
def get_request_json(self, *args, **kwargs):
|
||||
ret = super().get_request_json(*args, **kwargs)
|
||||
ret["image_data"] = [self.image_url]
|
||||
ret["logprob_start_len"] = (
|
||||
-1
|
||||
) # Do not try to calculate logprobs of image embeddings.
|
||||
ret[
|
||||
"logprob_start_len"
|
||||
] = -1 # Do not try to calculate logprobs of image embeddings.
|
||||
return ret
|
||||
|
||||
def test_simple_decode_stream(self):
|
||||
|
||||
Reference in New Issue
Block a user