loading more

tests/test_dynamic_routing.py

@@ -370,7 +370,11 @@ def test_admin_can_replace_a_served_model_and_release_it():
         headers = {"Content-Type": "application/json", "Authorization": "Bearer test-admin"}
         load = urllib.request.Request(
             f"http://127.0.0.1:{port}/v1/models/load",
-            data=json.dumps({"model": "qwen2.5-0.5b-instruct", "force": True}).encode(),
+            data=json.dumps({
+                "model": "qwen2.5-0.5b-instruct",
+                "node_id": node["node_id"],
+                "force": True,
+            }).encode(),
             headers=headers, method="POST")
         with urllib.request.urlopen(load) as response:
             assert json.loads(response.read())["assignment"]["node_id"] == node["node_id"]