Update src/interface.jl

Co-authored-by: Brian Chen <[email protected]>
FluxML · Oct 14, 2024 · 8ceedbe · 8ceedbe
1 parent e8dae73
commit 8ceedbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/interface.jl b/src/interface.jl
@@ -68,9 +68,9 @@ function update(tree, model, grad, higher...)
 end
 
 function update!(::AbstractRule, model, grad, higher...)
-  error("""update! must be called with an optimiser state, not a rule.
+  throw(ArgumentError("""update! must be called with an optimiser state tree, not a rule.
   Call `opt_state = setup(rule, model)` first, then `update!(opt_state, model, grad)`.
-  """)
+  """))
 end
 
 function update!(tree, model, grad, higher...)