From 6e7354e9a6c04c5bf3e24ddda9d76936f0b54964 Mon Sep 17 00:00:00 2001
From: Xwg <loloxwg@gmail.com>
Date: Mon, 13 Jan 2025 15:05:48 +0800
Subject: [PATCH] Fix output handling logic in graph.rs

Removed unnecessary conditional check for multiple outputs, streamlining the code. This change ensures consistent handling of output types and reduces complexity in the logic.
---
 crates/burn-import/src/burn/graph.rs | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

diff --git a/crates/burn-import/src/burn/graph.rs b/crates/burn-import/src/burn/graph.rs
index f6dee5479d..fbcc72bc07 100644
--- a/crates/burn-import/src/burn/graph.rs
+++ b/crates/burn-import/src/burn/graph.rs
@@ -472,15 +472,7 @@ impl<PS: PrecisionSettings> BurnGraph<PS> {
         self.graph_output_types.iter().for_each(|output| {
             let name = output.name();
             let ty = output.ty();
-
-            if multiple_output {
-                output_type_def.extend(quote! {
-                    #ty,
-                });
-                output_return_def.extend(quote! {
-                    #name,
-                });
-            } else {
+            {
                 output_type_def.extend(quote! {
                     #ty
                 });