diff --git a/src/MLJFlow.jl b/src/MLJFlow.jl index 7fd9a03..b99d7e5 100644 --- a/src/MLJFlow.jl +++ b/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") diff --git a/src/types.jl b/src/types.jl index 6a2f9d4..354bbc1 100644 --- a/src/types.jl +++ b/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 \ No newline at end of file diff --git a/test/types.jl b/test/types.jl index ba474ca..5814480 100644 --- a/test/types.jl +++ b/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