Fix type stability of Params.order

FluxML · Apr 13, 2022 · c560ed1 · c560ed1
1 parent 95e1021
commit c560ed1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/compiler/interface.jl b/src/compiler/interface.jl
@@ -138,8 +138,8 @@ gradient
 Container for implicit parameters, used when differentiating
 a zero-argument funtion `() -> loss(A, B)` with respect to `A, B`.
 """
-struct Params
-  order::Buffer # {Any, Vector{Any}}
+struct Params{B <: Buffer}
+  order::B
   params::IdSet{Any} # TODO store ids only
 end