anymutable should ignore fields that refer to parent type

FluxML · Feb 6, 2024 · 51ba0be · 51ba0be
1 parent 19788ea
commit 51ba0be
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/walks.jl b/src/walks.jl
@@ -112,9 +112,10 @@ usecache(::Union{AbstractDict, AbstractSet}, x) =
   isleaf(x) ? anymutable(x) : ismutable(x)
 usecache(::Nothing, x) = false
 
-@generated function anymutable(x::T) where {T}
+@generated function Functors.anymutable(x::T) where {T}
   ismutabletype(T) && return true
-  subs =  [:(anymutable(getfield(x, $f))) for f in QuoteNode.(fieldnames(T))]
+  fns = QuoteNode.(filter(n -> fieldtype(T, n) != T, fieldnames(T)))
+  subs =  [:(anymutable(getfield(x, $f))) for f in fns]
   return Expr(:(||), subs...)
 end