xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 11:06:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix][Failing Test] Fix test_events.py (#18460)

Browse Source 
Signed-off-by: rabi <ramishra@redhat.com>
This commit is contained in:
Rabi Mishra 2025-05-21 17:27:28 +05:30 committed by GitHub
parent 107f5fc4cb
commit 61acfc45bc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.buildkite/test-pipeline.yaml
View File

@ -138,6 +138,7 @@ steps:
- vllm/core/ - vllm/core/
- tests/distributed/test_utils - tests/distributed/test_utils
- tests/distributed/test_pynccl - tests/distributed/test_pynccl
- tests/distributed/test_events
- tests/spec_decode/e2e/test_integration_dist_tp4 - tests/spec_decode/e2e/test_integration_dist_tp4
- tests/compile/test_basic_correctness - tests/compile/test_basic_correctness
- examples/offline_inference/rlhf.py - examples/offline_inference/rlhf.py
@ -156,6 +157,7 @@ steps:
- pytest -v -s distributed/test_utils.py - pytest -v -s distributed/test_utils.py
- pytest -v -s compile/test_basic_correctness.py - pytest -v -s compile/test_basic_correctness.py
- pytest -v -s distributed/test_pynccl.py - pytest -v -s distributed/test_pynccl.py
- pytest -v -s distributed/test_events.py
- pytest -v -s spec_decode/e2e/test_integration_dist_tp4.py - pytest -v -s spec_decode/e2e/test_integration_dist_tp4.py
# TODO: create a dedicated test section for multi-GPU example tests # TODO: create a dedicated test section for multi-GPU example tests
# when we have multiple distributed example tests # when we have multiple distributed example tests

9
tests/distributed/test_events.py
View File

@ -119,13 +119,12 @@ def test_topic_filtering(publisher_config):
""" """
publisher_config.replay_endpoint = None publisher_config.replay_endpoint = None
cfg = publisher_config.model_copy() publisher_config.topic = "foo"
cfg.topic = "foo" pub = EventPublisherFactory.create(publisher_config)
pub = EventPublisherFactory.create(cfg)
from .conftest import MockSubscriber from .conftest import MockSubscriber
sub_foo = MockSubscriber(cfg.endpoint, None, "foo") sub_foo = MockSubscriber(publisher_config.endpoint, None, "foo")
sub_bar = MockSubscriber(cfg.endpoint, None, "bar") sub_bar = MockSubscriber(publisher_config.endpoint, None, "bar")
try: try:
time.sleep(0.1) time.sleep(0.1)