Skip to content

Commit 4aef98d

Browse files
fix(test): add missing colocate field to wu_controller connect tests (#1357)
1 parent 6a84d4e commit 4aef98d

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

tests/experimental/weight_update/test_wu_controller.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ def test_connect_sends_correct_request(self, ctrl):
8181
"save_path": "",
8282
"use_lora": False,
8383
"lora_name": "",
84+
"colocate": False,
8485
},
8586
timeout=10.0,
8687
)
@@ -110,6 +111,7 @@ def test_connect_disk_mode_sends_disk_fields(self, ctrl):
110111
"save_path": "/shared/weights",
111112
"use_lora": True,
112113
"lora_name": "my-lora",
114+
"colocate": False,
113115
},
114116
timeout=10.0,
115117
)

0 commit comments

Comments
 (0)