From 5370f0ef9fa2e5fd1fe5fc4318fee1d9187600ad Mon Sep 17 00:00:00 2001
From: Jose Esparza <28990958+pebeto@users.noreply.github.com>
Date: Thu, 17 Aug 2023 08:54:00 -0500
Subject: [PATCH] Adding verbosity field in MLFlowLogger type

---
 src/types.jl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/types.jl b/src/types.jl
index b1053ea..a74d79e 100644
--- a/src/types.jl
+++ b/src/types.jl
@@ -25,15 +25,16 @@ more information, see [MLFlowClient.jl](https://juliaai.github.io/MLFlowClient.j
 """
 struct MLFlowLogger
     service::MLFlow
+    verbosity::Integer
     experiment_name::String
     artifact_location::Union{String,Nothing}
 end
 function MLFlowLogger(baseuri; experiment_name="MLJ experiment",
-    artifact_location=nothing)
+    artifact_location=nothing, verbosity=1)
     service = MLFlow(baseuri)
 
     if ~healthcheck(service)
         error("It seems that the MLFlow server is not running. For more information, see https://mlflow.org/docs/latest/quickstart.html")
     end
-    MLFlowLogger(service, experiment_name, artifact_location)
+    MLFlowLogger(service, experiment_name, artifact_location, verbosity)
 end