Remove broadcast-related ambiguity with Params

FluxML · Dec 31, 2023 · 2cd611e · 2cd611e
1 parent c8fe70d
commit 2cd611e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/compiler/interface.jl b/src/compiler/interface.jl
@@ -257,7 +257,7 @@ function Base.delete!(ps::Params, x)
   return ps
 end
 
-Base.Broadcast.broadcasted(f, ps::Params) = broadcasted(f, ps.order)
+Base.Broadcast.broadcastable(ps::Params) = ps.order
 
 @adjoint function Broadcast.broadcasted(f::Function, ps::Params)
   f.(ps), _ -> throw(ArgumentError("Zygote.Params does not support broadcasting within gradients, try iteration `for p in ps`"))