From 26c7bdeab20b85f9e34d16f04f94b5b78985a412 Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Fri, 10 Feb 2023 15:35:01 +0100
Subject: [PATCH] formatting

---
 router/src/infer.rs | 6 ++++--
 router/src/main.rs  | 5 +----
 router/src/queue.rs | 3 ++-
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/router/src/infer.rs b/router/src/infer.rs
index 4013edfd..791a5b5e 100644
--- a/router/src/infer.rs
+++ b/router/src/infer.rs
@@ -244,10 +244,12 @@ async fn batching_task(
                 }
                 // Create span for this batch to add context to inference calls
                 let next_batch_size = entries.len();
-                let next_batch_span = info_span!(parent: None, "batch", batch_size = next_batch_size);
+                let next_batch_span =
+                    info_span!(parent: None, "batch", batch_size = next_batch_size);
                 entries.iter_mut().for_each(|(_, entry)| {
                     // Create a new span to link the batch back to this entry
-                    let entry_batch_span = info_span!(parent: &entry.span, "infer", batch_size = next_batch_size);
+                    let entry_batch_span =
+                        info_span!(parent: &entry.span, "infer", batch_size = next_batch_size);
                     // Add relationship
                     entry_batch_span.follows_from(&next_batch_span);
                     // Update entry
diff --git a/router/src/main.rs b/router/src/main.rs
index de67178f..881d94fb 100644
--- a/router/src/main.rs
+++ b/router/src/main.rs
@@ -118,10 +118,7 @@ fn init_logging(otlp_endpoint: Option<String>, json_output: bool) {
         .with_line_number(true);
 
     let fmt_layer = match json_output {
-        true => fmt_layer
-            .json()
-            .flatten_event(true)
-            .boxed(),
+        true => fmt_layer.json().flatten_event(true).boxed(),
         false => fmt_layer.boxed(),
     };
     layers.push(fmt_layer);
diff --git a/router/src/queue.rs b/router/src/queue.rs
index bba3b675..891e821a 100644
--- a/router/src/queue.rs
+++ b/router/src/queue.rs
@@ -158,7 +158,8 @@ impl State {
             .drain(..next_batch_size)
             .for_each(|(id, mut entry)| {
                 // Create a new span to link the batch back to this entry
-                let entry_batch_span = info_span!(parent: &entry.span, "infer", batch_size = next_batch_size);
+                let entry_batch_span =
+                    info_span!(parent: &entry.span, "infer", batch_size = next_batch_size);
                 // Add relationship
                 entry_batch_span.follows_from(&next_batch_span);
                 // Update entry