Skip to content

Commit

Permalink
fix tests
Browse files Browse the repository at this point in the history
Signed-off-by: Alexander Matveev <[email protected]>
  • Loading branch information
alexm-neuralmagic committed Dec 11, 2024
1 parent adda9d4 commit 074fef9
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 0 deletions.
1 change: 1 addition & 0 deletions tests/v1/engine/test_engine_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ def make_request() -> EngineCoreRequest:
prompt=PROMPT,
prompt_token_ids=PROMPT_TOKENS,
mm_inputs=None,
mm_hashes=None,
mm_placeholders=None,
sampling_params=SamplingParams(),
eos_token_id=None,
Expand Down
1 change: 1 addition & 0 deletions tests/v1/engine/test_engine_core_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ def make_request(params: SamplingParams) -> EngineCoreRequest:
prompt=PROMPT,
prompt_token_ids=PROMPT_TOKENS,
mm_inputs=None,
mm_hashes=None,
mm_placeholders=None,
sampling_params=params,
eos_token_id=None,
Expand Down

0 comments on commit 074fef9

Please sign in to comment.