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/burn-import/src/burn/node/depth_to_space.rs

@@ -127,7 +127,7 @@ mod tests {
 
         graph.register(DepthToSpaceNode::new(
             TensorType::new_float("input", 4),
-            TensorType::new_float("output", 5),
+            TensorType::new_float("output", 4),
             DepthToSpaceConfig::new(DepthToSpaceMode::CRD, 2),
         ));
 

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

@@ -44,7 +44,7 @@ pub fn depth_to_space_update_outputs(node: &mut Node) {
     node.outputs[0].ty = ArgType::Tensor(TensorType {
         elem_type: tensor.elem_type.clone(),
         rank: tensor.rank,
-        static_shape: static_shape,
+        static_shape,
     });
 }
 
@@ -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(

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

@@ -44,7 +44,7 @@ pub fn space_to_depth_update_outputs(node: &mut Node) {
     node.outputs[0].ty = ArgType::Tensor(TensorType {
         elem_type: tensor.elem_type.clone(),
         rank: tensor.rank,
-        static_shape: static_shape,
+        static_shape,
     });
 }