Merge pull request #32 from JuliaAI/24-improve-show-for-logger-instances

Extending Base.show to pretty print logger instances

src/MLJFlow.jl

@@ -6,6 +6,7 @@ using MLFlowClient: MLFlow, logparam, logmetric,
     createrun, MLFlowRun, updaterun,
     healthcheck, logartifact, getorcreateexperiment
 
+import Base: show
 import MLJBase: save, log_evaluation
 
 include("types.jl")

src/types.jl

@@ -39,3 +39,11 @@ function Logger(baseuri; experiment_name="MLJ experiment",
     end
     Logger(service, verbosity, experiment_name, artifact_location)
 end
+function show(io::IO, logger::MLJFlow.Logger)
+    print(io,
+        "MLFLowLogger(\"$(logger.service.baseuri)\",\n" *
+        "   experiment_name=\"$(logger.experiment_name)\",\n" *
+        "   artifact_location=\"$(logger.artifact_location)\",\n" *
+        ") using MLFlow API version $(logger.service.apiversion)"
+    )
+end

test/types.jl

@@ -3,4 +3,12 @@
 
     @test typeof(logger) == MLJFlow.Logger
     @test typeof(logger.service) == MLFlow
+
+    io = IOBuffer()
+    show(io, logger)
+    test_string = "MLFLowLogger(\"$(logger.service.baseuri)\",\n" *
+        "   experiment_name=\"$(logger.experiment_name)\",\n" *
+        "   artifact_location=\"$(logger.artifact_location)\",\n" *
+        ") using MLFlow API version $(logger.service.apiversion)"
+    @test String(take!(io)) == test_string
 end