fix ambiguity

src/Flux.jl

@@ -57,10 +57,8 @@ export Chain, Dense, Embedding, EmbeddingBag,
 ))
 
 include("optimise/Optimise.jl")
-using .Optimise: Optimise, update!
-export Optimise, ClipValue, Optimiser
-
-export ClipGrad, OptimiserChain  # these are const defined in deprecations, for ClipValue, Optimiser
+using .Optimise: Optimise
+export ClipValue # this is const defined in deprecations, for ClipGrad
 
 include("train.jl")
 using .Train

src/deprecations.jl

@@ -103,6 +103,11 @@ function update!(opt::Optimisers.AbstractRule, model, grad)
      `update!(state, model, grad)` needs `state = Flux.setup(opt, model)`.
     """)
 end
+function update!(opt::Optimisers.AbstractRule, model::Chain, grad::Tuple)
+  error("""Invalid input to `update!`.
+     `update!(state, model, grad)` needs `state = Flux.setup(opt, model)`.
+    """)
+end
 
 # An easy error to make is to pass result of explicit gradient(...), not gradient(...)[1]
 # Can't catch every case, but can catch many simple Flux models:

test/train.jl

@@ -90,7 +90,7 @@ end
 
 @testset "Explicit Flux.update! features" begin
   m = Chain(Dense(2=>3, tanh), Dense(3=>1), only)
-  x = rand(2)
+  x = rand(Float32, 2)
   y1 = m(x)  # before
 
   # Explicit gradient