From 06254d4cbb79e0a406fbf4e18d739293d5470114 Mon Sep 17 00:00:00 2001 From: Chauncey Date: Sat, 28 Feb 2026 14:47:43 +0800 Subject: [PATCH] [CI] add trainer_send_weights for MockWeightTransferEngine (#35589) Signed-off-by: chaunceyjiang --- tests/entrypoints/weight_transfer/test_weight_transfer_llm.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/entrypoints/weight_transfer/test_weight_transfer_llm.py b/tests/entrypoints/weight_transfer/test_weight_transfer_llm.py index cd13aca7e..255bca444 100644 --- a/tests/entrypoints/weight_transfer/test_weight_transfer_llm.py +++ b/tests/entrypoints/weight_transfer/test_weight_transfer_llm.py @@ -90,6 +90,10 @@ class MockWeightTransferEngine(WeightTransferEngine[MockInitInfo, MockUpdateInfo def shutdown(self) -> None: MockWeightTransferEngine.shutdown_called = True + def trainer_send_weights(self, *args, **kwargs): + """Mock method to simulate trainer sending weights.""" + pass + def mock_create_engine(config, parallel_config): """Mock factory function that returns our mock engine."""