Adding Makie extension for artifact logging

Author: rkube

Project.toml

@@ -15,9 +15,13 @@ ShowCases = "605ecd9f-84a6-4c9e-81e2-4798472b76a3"
 URIs = "5c2747f8-b7ea-4ff2-ba2e-563bfd36b1d4"
 UUIDs = "cf7118a7-6976-5b1a-9a39-7adc72f591a4"
 
+[weakdeps]
+CairoMakie = "13f3f980-e62b-5c42-98c6-ff1f3baf88f0"
+
 [compat]
 AWSS3 = "0.11.4"
 Base64 = "1.0"
+CairoMakie = "0.15.7"
 FileTypes = "0.1.1"
 HTTP = "1.0"
 JSON = "0.21"
@@ -32,3 +36,6 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
 test = ["Base64", "Test"]
+
+[extensions]
+MLFlowMakieExt = "CairoMakie"

ext/logger.jl

@@ -0,0 +1,45 @@
+
+module MLFlowMakieExt
+
+using CairoMakie
+
+"""
+    logartifact(s3_cfg::MinioConfig, run::Run, fig::Makie.Figure, artifact_name::String="")
+
+Log artifact for a run. Supports only S3 buckets.
+
+# Arguments
+- `s3_cfg`: Minio configuration
+- `Run`: ['Run'](@ref) instance
+- `fig`: `Makie.Figure` to store
+- `artifact_name`: Name of the artifact in the bucket
+
+# Returns
+`true` if successful. Otherwise, raises exception.
+
+"""
+
+function logartifact(s3_cfg::MinioConfig, run::Run, fig::Makie.Figure, artifact_name::String="")
+    # Parse the URI
+    u = URI(run.info.artifact_uri)
+    u.scheme == "s3" || ArgumentError("The artifact URI for the run has to be a S3 bucket. Got: $(run.info.artifact_uri)")
+    isfile(path) || ArgumentError("Can not read file $(path).")
+    bucket_name = u.host
+    artifacts_base_path = u.path
+
+    isempty(artifact_name) || ArgumentError("The provided artifact_name is empty")
+
+    # Store figure as png
+    io = IOBuffer()
+    show(io, MIME"figure/png"(), fig)
+    content = take!(io)
+
+    # Create artifact path
+    artifact_path = joinpath(artifacts_base_path, artifact_name)
+
+    # store in bucket
+    s3_put(s3_cfg, bucket_name, artifact_path, content, "figure/png")
+    return true
+end
+
+end

src/services/logger.jl

@@ -141,7 +141,7 @@ logmodel(instance::MLFlow, run::Run, model_json::String)::Bool =
     logmodel(instance, run.info.run_id, model_json)
 
 """
-    logartifact(s3_cfg::MinioConfig, run::Run, s3_cfg::MinioConfig, path::String="", artifact_name::String="")
+    logartifact(s3_cfg::MinioConfig, run::Run, path::String="", artifact_name::String="")
 
 Log artifact for a run. Supports only S3 buckets.
 
@@ -184,3 +184,5 @@ function logartifact(s3_cfg::MinioConfig, run::Run, path::String="", artifact_na
 end
 
 
+
+