[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
@@ -131,7 +131,6 @@ class MiniLoadBalancer:
|
||||
total=self.timeout
|
||||
) # Add timeout for request reliability
|
||||
) as session:
|
||||
|
||||
tasks = [
|
||||
session.post(f"{prefill_server}/{endpoint}", json=prefill_req),
|
||||
session.post(f"{decode_server}/{endpoint}", json=decode_req),
|
||||
@@ -141,7 +140,6 @@ class MiniLoadBalancer:
|
||||
prefill_response, decode_response = await asyncio.gather(*tasks)
|
||||
|
||||
if "return_logprob" in modified_request:
|
||||
|
||||
prefill_json = await prefill_response.json()
|
||||
ret_json = await decode_response.json()
|
||||
|
||||
|
||||
@@ -1069,7 +1069,6 @@ class RouterArgs:
|
||||
|
||||
prefill_urls = []
|
||||
for prefill_args in prefill_list:
|
||||
|
||||
url = prefill_args[0]
|
||||
|
||||
# Handle optional bootstrap port
|
||||
|
||||
@@ -788,7 +788,7 @@ def test_find_available_ports_and_wait_health(monkeypatch):
|
||||
monkeypatch.setattr(
|
||||
ls.time,
|
||||
"perf_counter",
|
||||
lambda: (base.__setitem__("t", base["t"] + 0.1) or base["t"]),
|
||||
lambda: base.__setitem__("t", base["t"] + 0.1) or base["t"],
|
||||
)
|
||||
|
||||
assert ls.wait_for_server_health("127.0.0.1", 12345, timeout=1)
|
||||
@@ -922,7 +922,6 @@ def test_launch_server_process_declares_on_a_resolved_record(monkeypatch):
|
||||
)
|
||||
|
||||
with patch("sglang_router.launch_router.logger") as mock_logger:
|
||||
|
||||
with pytest.raises(
|
||||
ValueError, match="PD disaggregation mode requires --prefill"
|
||||
):
|
||||
|
||||
Reference in New Issue
Block a user