Skip to content

Commit da7f4f4

Browse files
committed
Fix worker test stub missing sampling_params
Signed-off-by: Guan-Ming (Wesley) Chiu <105915352+guan404ming@users.noreply.github.com>
1 parent cef560b commit da7f4f4

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

tests/unit/scripts/test_response_regeneration.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import importlib.util
1212
import json
1313
from pathlib import Path
14+
from typing import Any
1415

1516
import pytest
1617

@@ -434,6 +435,7 @@ class _Args:
434435
model = "m"
435436
max_tokens = 16
436437
max_retries = 0
438+
sampling_params: dict[str, Any] = {}
437439

438440

439441
class _NullProgress:

0 commit comments

Comments
 (0)