fix: ci

Author: tye-singwa

crates/burn-import/src/burn/node/base.rs

@@ -401,9 +401,10 @@ pub(crate) mod tests {
             ),
         ));
 
-        graph.register_input_output(vec!["tensor1".to_string(), "tensor2".to_string()], vec![
-            "tensor4".to_string(),
-        ]);
+        graph.register_input_output(
+            vec!["tensor1".to_string(), "tensor2".to_string()],
+            vec!["tensor4".to_string()],
+        );
 
         let expected = quote! {
             use burn::{
@@ -486,9 +487,10 @@ pub(crate) mod tests {
             TensorType::new_float("output", 4),
         ));
 
-        graph.register_input_output(vec!["tensor1".to_string(), "tensor2".to_string()], vec![
-            "output".to_string(),
-        ]);
+        graph.register_input_output(
+            vec!["tensor1".to_string(), "tensor2".to_string()],
+            vec!["output".to_string()],
+        );
 
         let expected = quote! {
             use burn::{

crates/onnx-ir/src/node/depth_to_space.rs

@@ -104,9 +104,9 @@ pub fn depth_to_space_config(node: &Node) -> DepthToSpaceConfig {
 #[cfg(test)]
 mod tests {
     use super::*;
+    use crate::ElementType;
     use crate::ir::NodeType;
     use crate::node::test_utils::NodeBuilder;
-    use crate::ElementType;
 
     /// Helper function to create test nodes with different repeat values
     fn create_test_node(