diff --git a/tests/unittest/executor/test_rpc.py b/tests/unittest/executor/test_rpc.py index 8a04d7534c2..c9ba44db3d7 100644 --- a/tests/unittest/executor/test_rpc.py +++ b/tests/unittest/executor/test_rpc.py @@ -218,6 +218,7 @@ def task(self): client.close() + @pytest.mark.skip(reason="https://nvbugs/5579234") def test_timeout_error(self): """Test that requests that exceed timeout are handled with proper error.""" diff --git a/tests/unittest/executor/test_rpc_proxy.py b/tests/unittest/executor/test_rpc_proxy.py index 17d99fd24d7..94615615a09 100644 --- a/tests/unittest/executor/test_rpc_proxy.py +++ b/tests/unittest/executor/test_rpc_proxy.py @@ -55,6 +55,7 @@ def create_proxy(self, tp_size: int): return proxy + @pytest.mark.skip(reason="https://nvbugs/5579234") @pytest.mark.parametrize("num_reqs", [1, 10]) def test_tp1(self, num_reqs): tokenizer = TransformersTokenizer.from_pretrained(model_path)