Skip to content

Commit

Permalink
Fix output handling logic in graph.rs
Browse files Browse the repository at this point in the history
Removed unnecessary conditional check for multiple outputs, streamlining the code. This change ensures consistent handling of output types and reduces complexity in the logic.
  • Loading branch information
loloxwg committed Jan 13, 2025
1 parent da8de56 commit 6e7354e
Showing 1 changed file with 1 addition and 9 deletions.
10 changes: 1 addition & 9 deletions crates/burn-import/src/burn/graph.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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
});
Expand Down

0 comments on commit 6e7354e

Please sign in to comment.