Fix tests mod in queue.rs

huggingface · Mar 26, 2023 · 9895569 · 9895569
1 parent f934d01
commit 9895569
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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,