From 9895569c8be254a9e67db3c5d281724e9f02790b Mon Sep 17 00:00:00 2001
From: Nick Hill <nickhill@us.ibm.com>
Date: Sat, 25 Mar 2023 22:01:05 -0700
Subject: [PATCH] Fix tests mod in queue.rs

---
 router/src/queue.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/router/src/queue.rs b/router/src/queue.rs
index 132fe558..0b6c4bde 100644
--- a/router/src/queue.rs
+++ b/router/src/queue.rs
@@ -219,9 +219,10 @@ enum QueueCommand {
 mod tests {
     use super::*;
     use std::sync::Arc;
-    use text_generation_client::{NextTokenChooserParameters, StoppingCriteriaParameters};
+    use text_generation_client::NextTokenChooserParameters;
     use tokio::sync::{mpsc, Semaphore};
     use tracing::info_span;
+    use crate::validation::StoppingCriteria;
 
     fn default_entry() -> Entry {
         let semaphore = Arc::new(Semaphore::new(1));
@@ -241,9 +242,10 @@ mod tests {
                     repetition_penalty: 0.0,
                     watermark: false,
                 },
-                stopping_parameters: StoppingCriteriaParameters {
+                stopping_criteria: StoppingCriteria {
                     max_new_tokens: 0,
                     stop_sequences: vec![],
+                    deadline: None,
                 },
             },
             response_tx,