[model-gateway] optimize core modules (#14751)
This commit is contained in:
@@ -51,7 +51,7 @@ mod test_pd_routing {
|
||||
assert_eq!(prefill_worker.url(), "http://prefill:8080");
|
||||
match prefill_worker.worker_type() {
|
||||
WorkerType::Prefill { bootstrap_port } => {
|
||||
assert_eq!(bootstrap_port, Some(9000));
|
||||
assert_eq!(*bootstrap_port, Some(9000));
|
||||
}
|
||||
_ => panic!("Expected Prefill worker type"),
|
||||
}
|
||||
@@ -353,7 +353,7 @@ mod test_pd_routing {
|
||||
|
||||
let bootstrap_port = match prefill_worker.worker_type() {
|
||||
WorkerType::Prefill { bootstrap_port } => bootstrap_port,
|
||||
_ => None,
|
||||
_ => &None,
|
||||
};
|
||||
|
||||
single_json["bootstrap_host"] = json!(prefill_worker.bootstrap_host());
|
||||
@@ -697,7 +697,7 @@ mod test_pd_routing {
|
||||
|
||||
let bootstrap_port = match prefill_worker.worker_type() {
|
||||
WorkerType::Prefill { bootstrap_port } => bootstrap_port,
|
||||
_ => None,
|
||||
_ => &None,
|
||||
};
|
||||
let batch_size = 16;
|
||||
let hostname = prefill_worker.bootstrap_host();
|
||||
@@ -823,7 +823,7 @@ mod test_pd_routing {
|
||||
|
||||
let bootstrap_port = match prefill_worker.worker_type() {
|
||||
WorkerType::Prefill { bootstrap_port } => bootstrap_port,
|
||||
_ => None,
|
||||
_ => &None,
|
||||
};
|
||||
let hostname = prefill_worker.bootstrap_host();
|
||||
|
||||
|
||||
Reference in New Issue
Block a user