Skip to content

Commit c4c8ca8

Browse files
committed
fix formatting
Signed-off-by: Sophie du Couédic <[email protected]>
1 parent 48940a6 commit c4c8ca8

File tree

3 files changed

+8
-5
lines changed

3 files changed

+8
-5
lines changed

docs/contributing/continuous_batching/tests/e2e_testing.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,4 @@
77
options:
88
members:
99
- test_output
10-
- test_batch_handling
10+
- test_batch_handling

docs/contributing/continuous_batching/tests/other_tests.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@
33
!!! note
44
Unless otherwise specified, all the continuous batching tests are running with `max_model_len=256`
55

6-
::: tests.e2e.test_spyre_cb
6+
::: tests.e2e.test_spyre_cb

tests/e2e/test_spyre_cb_inference_steps.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
1-
"""Verification of the correctness of the step-by-step execution of continuous batching. It does so by comparing, at every engine step (i.e. prefill or decode iteration), a bunch of attributes. This allows a finer testing of the padding and scheduling implementation.
1+
"""Verification of the correctness of the step-by-step execution of continuous
2+
batching. It does so by comparing, at every engine step (i.e. prefill or decode
3+
iteration), a bunch of attributes. This allows a finer testing of the padding
4+
and scheduling implementation.
25
36
Run `python -m pytest tests/e2e/test_spyre_cb_inference_steps.py`.
47
"""
@@ -347,7 +350,7 @@ def two_sequences_finish_same_time_as_new_arrive(
347350
* 2: len = 30, max tokens = 30, step joining = 0
348351
* 3: len = 20, max tokens = 10, step joining = 31
349352
"""
350-
353+
351354
seqs_max_tokens = [30, 30, 10]
352355
prompts_lengths = [49, 30, 20]
353356
steps_add_reqs = [0, 0, 31]
@@ -483,7 +486,7 @@ def prompt_too_long_for_current_tkv(model: str, backend: str,
483486
* 1: len = 49, max tokens = 57, step joining = 0
484487
* 2: len = 70, max tokens = 67, step joining = 0
485488
"""
486-
489+
487490
seqs_max_tokens = [57, 67]
488491
prompts_lengths = [49, 70]
489492
steps_add_reqs = [0, 0]

0 commit comments

Comments
 (0)