From 2c3772528d735fd5841f1a1a1da94edd4cd4cea6 Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Fri, 20 Oct 2023 17:15:33 +0200
Subject: [PATCH] fix tests

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

diff --git a/router/src/queue.rs b/router/src/queue.rs
index 97159ef5..bbb8db0e 100644
--- a/router/src/queue.rs
+++ b/router/src/queue.rs
@@ -321,9 +321,9 @@ mod tests {
 
     fn default_entry() -> (
         Entry,
-        flume::Receiver<Result<InferStreamResponse, InferError>>,
+        mpsc::UnboundedReceiver<Result<InferStreamResponse, InferError>>,
     ) {
-        let (response_tx, receiver_tx) = flume::unbounded();
+        let (response_tx, receiver_tx) = mpsc::unbounded_channel();
 
         let entry = Entry {
             request: ValidGenerateRequest {