We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aabca47 commit 21e9db7Copy full SHA for 21e9db7
tests/worker/tpu_worker_test.py
@@ -25,6 +25,8 @@ def mock_vllm_config():
25
mock_parallel_conf = MagicMock()
26
mock_parallel_conf.tensor_parallel_size = 2
27
mock_parallel_conf.data_parallel_size = 1
28
+ mock_parallel_conf.nnodes = 1
29
+ mock_parallel_conf.nnodes_within_dp = 1
30
31
mock_additional_config = {}
32
0 commit comments