Skip to content

Commit cf69ecc

Browse files
committed
make metrics a dictionary instead of a vector
1 parent 749548d commit cf69ecc

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

src/types.jl

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ Base.show(io::IO, t::MLFlowRunDataMetric) = show(io, ShowCase(t, new_lines=true)
168168
Represents run data.
169169
170170
# Fields
171-
- `metrics::Vector{MLFlowRunDataMetric}`: run metrics.
171+
- `metrics::Dict{String,MLFlowRunDataMetric}`: run metrics.
172172
- `params::Dict{String,String}`: run parameters.
173173
- `tags`: list of run tags.
174174
@@ -178,12 +178,18 @@ Represents run data.
178178
179179
"""
180180
struct MLFlowRunData
181-
metrics::Vector{MLFlowRunDataMetric}
181+
metrics::Dict{String,MLFlowRunDataMetric}
182182
params::Union{Dict{String,String},Missing}
183183
tags
184184
end
185185
function MLFlowRunData(data::Dict{String,Any})
186-
metrics = haskey(data, "metrics") ? MLFlowRunDataMetric.(data["metrics"]) : MLFlowRunDataMetric[]
186+
metrics = Dict{String,MLFlowRunDataMetric}()
187+
if haskey(data, "metrics")
188+
for metric in data["metrics"]
189+
v = MLFlowRunDataMetric(metric)
190+
metrics[v.key] = v
191+
end
192+
end
187193
if haskey(data, "params")
188194
params = Dict{String,String}()
189195
for p in data["params"]
@@ -263,4 +269,4 @@ struct MLFlowArtifactDirInfo
263269
end
264270
Base.show(io::IO, t::MLFlowArtifactDirInfo) = show(io, ShowCase(t, new_lines=true))
265271
get_path(mlfadi::MLFlowArtifactDirInfo) = mlfadi.dirpath
266-
get_size(mlfadi::MLFlowArtifactDirInfo) = 0
272+
get_size(mlfadi::MLFlowArtifactDirInfo) = 0

0 commit comments

Comments
 (0)