Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
06254d4c
Unverified
Commit
06254d4c
authored
Feb 28, 2026
by
Chauncey
Committed by
GitHub
Feb 28, 2026
Browse files
[CI] add trainer_send_weights for MockWeightTransferEngine (#35589)
Signed-off-by:
chaunceyjiang
<
chaunceyjiang@gmail.com
>
parent
f5d1281c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
tests/entrypoints/weight_transfer/test_weight_transfer_llm.py
...s/entrypoints/weight_transfer/test_weight_transfer_llm.py
+4
-0
No files found.
tests/entrypoints/weight_transfer/test_weight_transfer_llm.py
View file @
06254d4c
...
...
@@ -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."""
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment