Unverified Commit 06254d4c authored by Chauncey's avatar Chauncey Committed by GitHub
Browse files

[CI] add trainer_send_weights for MockWeightTransferEngine (#35589)


Signed-off-by: default avatarchaunceyjiang <chaunceyjiang@gmail.com>
parent f5d1281c
......@@ -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."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment