Skip to content

Commit 21e9db7

Browse files
authored
[CI] Fix tpu worker test (#1124)
1 parent aabca47 commit 21e9db7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/worker/tpu_worker_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@ def mock_vllm_config():
2525
mock_parallel_conf = MagicMock()
2626
mock_parallel_conf.tensor_parallel_size = 2
2727
mock_parallel_conf.data_parallel_size = 1
28+
mock_parallel_conf.nnodes = 1
29+
mock_parallel_conf.nnodes_within_dp = 1
2830

2931
mock_additional_config = {}
3032

0 commit comments

Comments
 (0)