fixed deprecation warning from improper Vararg use (#145)

FluxML · Feb 28, 2023 · 8e5bfe8 · 8e5bfe8 · lorenzoh · Feb 28, 2023
1 parent 95f8177
commit 8e5bfe8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "FluxTraining"
 uuid = "7bf95e4d-ca32-48da-9824-f0dc5310474f"
 authors = ["lorenzoh <[email protected]>"]
-version = "0.3.6"
+version = "0.3.7"
 
 [deps]
 BSON = "fbb218c0-5317-5bc6-957e-2ee96dd4b1f0"

diff --git a/src/learner.jl b/src/learner.jl
@@ -74,11 +74,12 @@ function Learner(model, lossfn; callbacks = [], data = (), optimizer = ADAM(), k
 end
 
 
+# note that Vararg is covariant, using Vararg{<:Callback} produces warning
 """
     Learner(model, data, optimizer, lossfn, [callbacks...; kwargs...])
 """
 function Learner(
-        model, data, optimizer, lossfn, callbacks::Vararg{<:Callback};
+        model, data, optimizer, lossfn, callbacks::Vararg{Callback};
         usedefaultcallbacks=true, cbrunner=LinearRunner()
     )
     callbacks = collect(Callback, callbacks)