Skip to content

Commit d6986e5

Browse files
committed
cleanup
1 parent dbc6dc8 commit d6986e5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/v1/e2e/test_eplb_offline.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,9 @@ def test_eplb_model(
3333
)
3434

3535
# Save EPLB statistics to disk
36-
eplb_config_save = EPLBConfig(window_size=8,
37-
step_interval=10, save_load_window=True, save_dir="/tmp")
36+
eplb_config_save = EPLBConfig(
37+
window_size=8, step_interval=10, save_load_window=True, save_dir="/tmp"
38+
)
3839
llm = LLM(eplb_config=eplb_config_save, **llm_args)
3940
llm.generate(test_prompt)
4041
del llm

0 commit comments

Comments
 (0)