Merge pull request #38 from JuliaAI/save-hot-fix

Save hot fix
JuliaAI · Mar 8, 2024 · ab7442d · ab7442d
2 parents f1652ec + b3cfca0
commit ab7442d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "MLJFlow"
 uuid = "7b7b8358-b45c-48ea-a8ef-7ca328ad328f"
 authors = ["Jose Esparza <joseesparzadc@icloud.com>"]
-version = "0.4.0"
+version = "0.4.1"
 
 [deps]
 MLFlowClient = "64a0f543-368b-4a9a-827a-e71edb2a0b83"

diff --git a/src/base.jl b/src/base.jl
@@ -19,7 +19,7 @@ function log_evaluation(logger::Logger, performance_evaluation)
     updaterun(logger.service, run, "FINISHED")
 end
 
-function save(logger, machine:: Machine)
+function save(logger::Logger, machine:: Machine)
     io = IOBuffer()
     save(io, machine)
     seekstart(io)