From 3a4b52fba42acda996a9402e019274c5c7617e3d Mon Sep 17 00:00:00 2001 From: sitabulaixizawaluduo Date: Thu, 21 May 2026 17:06:30 +0800 Subject: [PATCH] fix(test): add missing colocate field to wu_controller connect tests --- tests/experimental/weight_update/test_wu_controller.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/experimental/weight_update/test_wu_controller.py b/tests/experimental/weight_update/test_wu_controller.py index 302d9356b7..85c69a8dac 100644 --- a/tests/experimental/weight_update/test_wu_controller.py +++ b/tests/experimental/weight_update/test_wu_controller.py @@ -81,6 +81,7 @@ def test_connect_sends_correct_request(self, ctrl): "save_path": "", "use_lora": False, "lora_name": "", + "colocate": False, }, timeout=10.0, ) @@ -110,6 +111,7 @@ def test_connect_disk_mode_sends_disk_fields(self, ctrl): "save_path": "/shared/weights", "use_lora": True, "lora_name": "my-lora", + "colocate": False, }, timeout=10.0, )