From 81656bd016ef795441088e13d2724a23fc504c47 Mon Sep 17 00:00:00 2001
From: drbh <david.richard.holtz@gmail.com>
Date: Tue, 22 Apr 2025 17:07:28 +0000
Subject: [PATCH] fix: adjust imports

---
 backends/v2/src/queue.rs | 2 +-
 backends/v3/src/queue.rs | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/backends/v2/src/queue.rs b/backends/v2/src/queue.rs
index 788ecee6..b5ec0742 100644
--- a/backends/v2/src/queue.rs
+++ b/backends/v2/src/queue.rs
@@ -12,7 +12,6 @@ use text_generation_router::validation::{
 use tokio::sync::{mpsc, oneshot};
 use tokio::time::Instant;
 use tracing::{info_span, instrument, Span};
-use std::collections::HashMap;
 
 /// Queue entry
 #[derive(Debug)]
@@ -402,6 +401,7 @@ impl From<ValidStoppingParameters> for StoppingCriteriaParameters {
 #[cfg(test)]
 mod tests {
     use super::*;
+    use std::collections::HashMap;
     use std::sync::Arc;
     use tracing::info_span;
 
diff --git a/backends/v3/src/queue.rs b/backends/v3/src/queue.rs
index fe8963b8..6cd2fd76 100644
--- a/backends/v3/src/queue.rs
+++ b/backends/v3/src/queue.rs
@@ -16,7 +16,6 @@ use text_generation_router::validation::{
 use tokio::sync::{mpsc, oneshot};
 use tokio::time::Instant;
 use tracing::{info_span, instrument, Instrument, Span};
-use std::collections::HashMap;
 
 /// Queue entry
 #[derive(Debug)]