Merge pull request #1203 from ancapdev/params-typestability

Fix type stability of Params.order
FluxML · Apr 13, 2022 · 1eb80c5 · 1eb80c5
2 parents 95e1021 + c560ed1
commit 1eb80c5
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