From 723c9d1a2ebed7bae95bf9e283520832e4f255f9 Mon Sep 17 00:00:00 2001
From: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
Date: Thu, 8 Aug 2024 16:24:51 -0400
Subject: [PATCH] Fix module derive with generics (#2127)

* Remove unnecessary ModuleDisplayDefault generic bound + duplicate ModuleDisplay

* Remove erroneous bound for autodiff module generic
---
 crates/burn-derive/src/module/codegen.rs | 25 ------------------------
 1 file changed, 25 deletions(-)

diff --git a/crates/burn-derive/src/module/codegen.rs b/crates/burn-derive/src/module/codegen.rs
index a6ab07efb..e91e69fb7 100644
--- a/crates/burn-derive/src/module/codegen.rs
+++ b/crates/burn-derive/src/module/codegen.rs
@@ -199,19 +199,12 @@ impl GenericsParser {
                 }
             );
 
-            module.add_predicate(
-                parse_quote! {
-                    #ident: burn::module::ModuleDisplayDefault
-                }
-            );
-
             module.add_predicate(
                 parse_quote! {
                     #ident: burn::module::ModuleDisplay
                 }
             );
 
-
             module_autodiff.add_predicate(
                 parse_quote! {
                     #ident: burn::module::AutodiffModule<B>
@@ -230,25 +223,7 @@ impl GenericsParser {
                 }
             );
 
-            module_autodiff.add_predicate(
-                parse_quote! {
-                    <#ident as burn::module::AutodiffModule<B>>::InnerModule: burn::module::ModuleDisplay
-                }
-            );
-
-
             generics_names_except_backend.extend(quote! { <#ident as burn::module::AutodiffModule<B>>::InnerModule, });
-            module_autodiff.add_predicate(
-                parse_quote! {
-                    #ident: burn::module::Module<B::InnerBackend>
-                }
-            );
-
-            module_autodiff.add_predicate(
-                parse_quote! {
-                    #ident: burn::module::ModuleDisplayDefault
-                }
-            );
 
             module_autodiff.add_predicate(
                 parse_quote! {