Skip to content

Commit ff8bd99

Browse files
pebetopebeto
committed
Applying formatting from Julia LSP
1 parent 7067af0 commit ff8bd99

22 files changed

+98
-98
lines changed

‎Project.toml

-1
Original file line numberDiff line numberDiff line change
@@ -23,4 +23,3 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
2323

2424
[targets]
2525
test = ["Test"]
26-

‎docs/make.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
push!(LOAD_PATH,"../src/")
1+
push!(LOAD_PATH, "../src/")
22
using Documenter
33
using MLFlowClient
44

‎src/api.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ MLFlowClient.uri(mlf, "experiments/get", Dict(:experiment_id=>10))
99
```
1010
"""
1111
uri(mlf::MLFlow, endpoint::String;
12-
parameters::Dict{Symbol, <:Any}=Dict{Symbol, NumberOrString}()) =
12+
parameters::Dict{Symbol,<:Any}=Dict{Symbol,NumberOrString}()) =
1313
URI("$(mlf.apiroot)/$(mlf.apiversion)/mlflow/$(endpoint)"; query=parameters)
1414

1515
"""

‎src/services/artifact.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ List artifacts for a [`Run`](@ref).
1717
- Token that can be used to retrieve the next page of artifact results.
1818
"""
1919
function listartifacts(instance::MLFlow, run_id::String; path::String="",
20-
page_token::String="")::Tuple{String, Array{FileInfo}, Union{String, Nothing}}
20+
page_token::String="")::Tuple{String,Array{FileInfo},Union{String,Nothing}}
2121
result = mlfget(instance, "artifacts/list"; run_id=run_id, path=path,
2222
page_token=page_token)
2323

‎src/services/experiment.jl

+2-2
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ fails if another [`Experiment`](@ref) with the same name already exists.
1919
The ID of the newly created [`Experiment`](@ref).
2020
"""
2121
function createexperiment(instance::MLFlow, name::String;
22-
artifact_location::Union{String, Missing}=missing,
22+
artifact_location::Union{String,Missing}=missing,
2323
tags::MLFlowUpsertData{Tag}=Tag[])::String
2424
result = mlfpost(instance, "experiments/create"; name=name,
2525
artifact_location=artifact_location, tags=parse(Tag, tags))
@@ -164,7 +164,7 @@ updateexperiment(instance::MLFlow, experiment::Experiment, new_name::String)::Bo
164164
"""
165165
function searchexperiments(instance::MLFlow; max_results::Int64=20000,
166166
page_token::String="", filter::String="", order_by::Array{String}=String[],
167-
view_type::ViewType=ACTIVE_ONLY)::Tuple{Array{Experiment}, Union{String, Nothing}}
167+
view_type::ViewType=ACTIVE_ONLY)::Tuple{Array{Experiment},Union{String,Nothing}}
168168
parameters = (; max_results, page_token, filter, :view_type => view_type |> Integer)
169169

170170
if order_by |> !isempty

‎src/services/logger.jl

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,14 +23,14 @@ represent ML model accuracy. A [`Metric`](@ref) can be logged multiple times.
2323
"""
2424
function logmetric(instance::MLFlow, run_id::String, key::String, value::Float64;
2525
timestamp::Int64=round(Int, now() |> datetime2unix),
26-
step::Union{Int64, Missing}=missing)::Bool
26+
step::Union{Int64,Missing}=missing)::Bool
2727
mlfpost(instance, "runs/log-metric"; run_id=run_id, key=key, value=value,
2828
timestamp=timestamp, step=step)
2929
return true
3030
end
3131
logmetric(instance::MLFlow, run::Run, key::String, value::Float64;
3232
timestamp::Int64=round(Int, now() |> datetime2unix),
33-
step::Union{Int64, Missing}=missing)::Bool =
33+
step::Union{Int64,Missing}=missing)::Bool =
3434
logmetric(instance, run.info.run_id, key, value; timestamp=timestamp, step=step)
3535
logmetric(instance::MLFlow, run_id::String, metric::Metric)::Bool =
3636
logmetric(instance, run_id, metric.key, metric.value, timestamp=metric.timestamp,

‎src/services/misc.jl

+6-6
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ Get a list of all values for the specified [`Metric`](@ref) for a given [`Run`](
1818
- The next page token if there are more results.
1919
"""
2020
function getmetrichistory(instance::MLFlow, run_id::String, metric_key::String;
21-
page_token::String="", max_results::Union{Int64, Missing}=missing
22-
)::Tuple{Array{Metric}, Union{String, Nothing}}
21+
page_token::String="", max_results::Union{Int64,Missing}=missing
22+
)::Tuple{Array{Metric},Union{String,Nothing}}
2323
result = mlfget(instance, "metrics/get-history"; run_id=run_id, metric_key=metric_key,
2424
page_token=page_token,
2525
max_results=(ismissing(max_results) ? max_results : (max_results |> Int32)))
@@ -30,13 +30,13 @@ function getmetrichistory(instance::MLFlow, run_id::String, metric_key::String;
3030
return metrics, next_page_token
3131
end
3232
getmetrichistory(instance::MLFlow, run::Run, metric_key::String; page_token::String="",
33-
max_results::Union{Int64, Missing}=missing
34-
)::Tuple{Array{Metric}, Union{String, Nothing}} =
33+
max_results::Union{Int64,Missing}=missing
34+
)::Tuple{Array{Metric},Union{String,Nothing}} =
3535
getmetrichistory(instance, run.info.run_id, metric_key; page_token=page_token,
3636
max_results=max_results)
3737
getmetrichistory(instance::MLFlow, run::Run, metric::Metric; page_token::String="",
38-
max_results::Union{Int64, Missing}=missing
39-
)::Tuple{Array{Metric}, Union{String, Nothing}} =
38+
max_results::Union{Int64,Missing}=missing
39+
)::Tuple{Array{Metric},Union{String,Nothing}} =
4040
getmetrichistory(instance, run.info.run_id, metric.key; page_token=page_token,
4141
max_results=max_results)
4242

‎src/services/model_version.jl

+5-5
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ end
3535
[`ModelVersion`](@ref) created.
3636
"""
3737
function createmodelversion(instance::MLFlow, name::String, source::String;
38-
run_id::Union{String, Missing}=missing, tags::MLFlowUpsertData{Tag}=Tag[],
39-
run_link::Union{String, Missing}=missing,
40-
description::Union{String, Missing}=missing)::ModelVersion
38+
run_id::Union{String,Missing}=missing, tags::MLFlowUpsertData{Tag}=Tag[],
39+
run_link::Union{String,Missing}=missing,
40+
description::Union{String,Missing}=missing)::ModelVersion
4141
result = mlfpost(instance, "model-versions/create"; name=name, source=source,
4242
run_id=run_id, tags=parse(Tag, tags), run_link=run_link, description=description)
4343
return result["model_version"] |> ModelVersion
@@ -73,7 +73,7 @@ end
7373
[`ModelVersion`](@ref) generated for this model in registry.
7474
"""
7575
function updatemodelversion(instance::MLFlow, name::String, version::String;
76-
description::Union{String, Missing}=missing)::ModelVersion
76+
description::Union{String,Missing}=missing)::ModelVersion
7777
result = mlfpatch(instance, "model-versions/update"; name=name, version=version,
7878
description=description)
7979
return result["model_version"] |> ModelVersion
@@ -114,7 +114,7 @@ end
114114
"""
115115
function searchmodelversions(instance::MLFlow; filter::String="",
116116
max_results::Int64=200000, order_by::Array{String}=String[],
117-
page_token::String="")::Tuple{Array{ModelVersion}, Union{String, Nothing}}
117+
page_token::String="")::Tuple{Array{ModelVersion},Union{String,Nothing}}
118118
parameters = (; max_results, page_token, filter)
119119

120120
if order_by |> !isempty

‎src/services/registered_model.jl

+3-3
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ An instance of type [`RegisteredModel`](@ref).
1818
"""
1919
function createregisteredmodel(instance::MLFlow, name::String;
2020
tags::MLFlowUpsertData{Tag}=Tag[],
21-
description::Union{String, Missing}=missing)::RegisteredModel
21+
description::Union{String,Missing}=missing)::RegisteredModel
2222
result = mlfpost(instance, "registered-models/create"; name=name,
2323
tags=parse(Tag, tags), description=description)
2424
return result["registered_model"] |> RegisteredModel
@@ -69,7 +69,7 @@ end
6969
An instance of type [`RegisteredModel`](@ref).
7070
"""
7171
function updateregisteredmodel(instance::MLFlow, name::String;
72-
description::Union{String, Missing}=missing)::RegisteredModel
72+
description::Union{String,Missing}=missing)::RegisteredModel
7373
result = mlfpatch(instance, "registered-models/update"; name=name,
7474
description=description)
7575
return result["registered_model"] |> RegisteredModel
@@ -109,7 +109,7 @@ end
109109
"""
110110
function searchregisteredmodels(instance::MLFlow; filter::String="",
111111
max_results::Int64=100, order_by::Array{String}=String[],
112-
page_token::String="")::Tuple{Array{RegisteredModel}, Union{String, Nothing}}
112+
page_token::String="")::Tuple{Array{RegisteredModel},Union{String,Nothing}}
113113
parameters = (; max_results, page_token, filter)
114114

115115
if order_by |> !isempty

‎src/services/run.jl

+10-9
Original file line numberDiff line numberDiff line change
@@ -18,19 +18,19 @@ single execution of a machine learning or data ETL pipeline.
1818
An instance of type [`Run`](@ref).
1919
"""
2020
function createrun(instance::MLFlow, experiment_id::String;
21-
run_name::Union{String, Missing}=missing, start_time::Union{Int64, Missing}=missing,
21+
run_name::Union{String,Missing}=missing, start_time::Union{Int64,Missing}=missing,
2222
tags::MLFlowUpsertData{Tag}=Tag[])::Run
2323
result = mlfpost(instance, "runs/create"; experiment_id=experiment_id,
2424
run_name=run_name, start_time=start_time, tags=parse(Tag, tags))
2525
return result["run"] |> Run
2626
end
2727
createrun(instance::MLFlow, experiment_id::Integer;
28-
run_name::Union{String, Missing}=missing, start_time::Union{Integer, Missing}=missing,
28+
run_name::Union{String,Missing}=missing, start_time::Union{Integer,Missing}=missing,
2929
tags::MLFlowUpsertData{Tag}=Tag[])::Run =
3030
createrun(instance, string(experiment_id); run_name=run_name, start_time=start_time,
3131
tags=tags)
3232
createrun(instance::MLFlow, experiment::Experiment;
33-
run_name::Union{String, Missing}=missing, start_time::Union{Integer, Missing}=missing,
33+
run_name::Union{String,Missing}=missing, start_time::Union{Integer,Missing}=missing,
3434
tags::MLFlowUpsertData{Tag}=Tag[])::Run =
3535
createrun(instance, string(experiment.experiment_id); run_name=run_name,
3636
start_time=start_time, tags=tags)
@@ -111,7 +111,8 @@ Set a [`Tag`](@ref) on a [`Run`](@ref).
111111
# Returns
112112
`true` if successful. Otherwise, raises exception.
113113
"""
114-
function setruntag(instance::MLFlow, run_id::String, key::String, value::String):Bool
114+
function setruntag(instance::MLFlow, run_id::String, key::String, value::String)
115+
:Bool
115116
mlfpost(instance, "runs/set-tag"; run_id=run_id, key=key, value=value)
116117
return true
117118
end
@@ -171,7 +172,7 @@ Search for runs that satisfy expressions. Search expressions can use [`Metric`](
171172
function searchruns(instance::MLFlow; experiment_ids::Array{String}=String[],
172173
filter::String="", run_view_type::ViewType=ACTIVE_ONLY, max_results::Int=1000,
173174
order_by::Array{String}=String[],
174-
page_token::String="")::Tuple{Array{Run}, Union{String, Nothing}}
175+
page_token::String="")::Tuple{Array{Run},Union{String,Nothing}}
175176
parameters = (; experiment_ids, filter, :run_view_type => run_view_type |> Integer,
176177
max_results, page_token)
177178

@@ -206,13 +207,13 @@ Update [`Run`](@ref) metadata.
206207
- An instance of type [`RunInfo`](@ref) with the updated metadata.
207208
"""
208209
function updaterun(instance::MLFlow, run_id::String;
209-
status::Union{RunStatus, Missing}=missing, end_time::Union{Int64, Missing}=missing,
210-
run_name::Union{String, Missing})::RunInfo
210+
status::Union{RunStatus,Missing}=missing, end_time::Union{Int64,Missing}=missing,
211+
run_name::Union{String,Missing})::RunInfo
211212
result = mlfpost(instance, "runs/update"; run_id=run_id, status=(status |> Integer),
212213
end_time=end_time, run_name=run_name)
213214
return result["run_info"] |> RunInfo
214215
end
215-
updaterun(instance::MLFlow, run::Run; status::Union{RunStatus, Missing}=missing,
216-
end_time::Union{Int64, Missing}=missing, run_name::Union{String, Missing})::RunInfo =
216+
updaterun(instance::MLFlow, run::Run; status::Union{RunStatus,Missing}=missing,
217+
end_time::Union{Int64,Missing}=missing, run_name::Union{String,Missing})::RunInfo =
217218
updaterun(instance, run.info.run_id; status=status, end_time=end_time,
218219
run_name=run_name)

‎src/types/dataset.jl

+4-4
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,10 @@ struct Dataset
1717
digest::String
1818
source_type::String
1919
source::String
20-
schema::Union{String, Nothing}
21-
profile::Union{String, Nothing}
20+
schema::Union{String,Nothing}
21+
profile::Union{String,Nothing}
2222
end
23-
Dataset(data::Dict{String, Any}) = Dataset(data["name"], data["digest"],
23+
Dataset(data::Dict{String,Any}) = Dataset(data["name"], data["digest"],
2424
data["source_type"], data["source"], get(data, "schema", nothing),
2525
get(data, "profile", nothing))
2626
Base.show(io::IO, t::Dataset) = show(io, ShowCase(t, new_lines=true))
@@ -38,6 +38,6 @@ struct DatasetInput
3838
tags::Array{Tag}
3939
dataset::Dataset
4040
end
41-
DatasetInput(data::Dict{String, Any}) = DatasetInput(
41+
DatasetInput(data::Dict{String,Any}) = DatasetInput(
4242
[Tag(tag) for tag in get(data, "tags", [])], Dataset(data["dataset"]))
4343
Base.show(io::IO, t::DatasetInput) = show(io, ShowCase(t, new_lines=true))

‎src/types/enums.jl

+13-13
Original file line numberDiff line numberDiff line change
@@ -8,11 +8,11 @@
88
- `READY`: Model version is ready for use.
99
"""
1010
@enum ModelVersionStatus begin
11-
PENDING_REGISTRATION=1
12-
FAILED_REGISTRATION=2
13-
READY=3
11+
PENDING_REGISTRATION = 1
12+
FAILED_REGISTRATION = 2
13+
READY = 3
1414
end
15-
ModelVersionStatus(status::String) = Dict(value => key for (key, value) in ModelVersionStatus |> Base.Enums.namemap)[status |> Symbol] |> ModelVersionStatus
15+
ModelVersionStatus(status::String) = Dict(value => key for (key, value) in ModelVersionStatus |> Base.Enums.namemap)[status|>Symbol] |> ModelVersionStatus
1616

1717
"""
1818
RunStatus
@@ -27,13 +27,13 @@ Status of a run.
2727
- `KILLED`: Run killed by user.
2828
"""
2929
@enum RunStatus begin
30-
RUNNING=1
31-
SCHEDULED=2
32-
FINISHED=3
33-
FAILED=4
34-
KILLED=5
30+
RUNNING = 1
31+
SCHEDULED = 2
32+
FINISHED = 3
33+
FAILED = 4
34+
KILLED = 5
3535
end
36-
RunStatus(status::String) = Dict(value => key for (key, value) in RunStatus |> Base.Enums.namemap)[status |> Symbol] |> RunStatus
36+
RunStatus(status::String) = Dict(value => key for (key, value) in RunStatus |> Base.Enums.namemap)[status|>Symbol] |> RunStatus
3737

3838
"""
3939
ViewType
@@ -46,7 +46,7 @@ View type for ListExperiments query.
4646
- `ALL`: Get all experiments.
4747
"""
4848
@enum ViewType begin
49-
ACTIVE_ONLY=1
50-
DELETED_ONLY=2
51-
ALL=3
49+
ACTIVE_ONLY = 1
50+
DELETED_ONLY = 2
51+
ALL = 3
5252
end

‎src/types/experiment.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ struct Experiment
2020
creation_time::Int64
2121
tags::Array{Tag}
2222
end
23-
Experiment(data::Dict{String, Any}) = Experiment(data["experiment_id"], data["name"],
23+
Experiment(data::Dict{String,Any}) = Experiment(data["experiment_id"], data["name"],
2424
data["artifact_location"], data["lifecycle_stage"], data["last_update_time"],
2525
data["creation_time"], [Tag(tag) for tag in get(data, "tags", [])])
2626
Base.show(io::IO, t::Experiment) = show(io, ShowCase(t, new_lines=true))

‎src/types/mlflow.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,6 @@ MLFlow(; apiroot="http://localhost:5000/api", apiversion=2.0, headers=Dict()) =
3232
ENV["MLFLOW_TRACKING_URI"] : apiroot), apiversion, headers)
3333

3434
Base.show(io::IO, t::MLFlow) =
35-
show(io, ShowCase(t, [:apiroot,:apiversion], new_lines=true))
35+
show(io, ShowCase(t, [:apiroot, :apiversion], new_lines=true))
3636

3737
abstract type LoggingData end

‎src/types/model_version.jl

+3-3
Original file line numberDiff line numberDiff line change
@@ -27,18 +27,18 @@ struct ModelVersion
2727
version::String
2828
creation_timestamp::Int64
2929
last_updated_timestamp::Int64
30-
user_id::Union{String, Nothing}
30+
user_id::Union{String,Nothing}
3131
current_stage::String
3232
description::String
3333
source::String
3434
run_id::String
3535
status::ModelVersionStatus
36-
status_message::Union{String, Nothing}
36+
status_message::Union{String,Nothing}
3737
tags::Array{Tag}
3838
run_link::String
3939
aliases::Array{String}
4040
end
41-
ModelVersion(data::Dict{String, Any}) = ModelVersion(data["name"], data["version"],
41+
ModelVersion(data::Dict{String,Any}) = ModelVersion(data["name"], data["version"],
4242
data["creation_timestamp"], data["last_updated_timestamp"],
4343
get(data, "user_id", nothing), data["current_stage"], data["description"],
4444
data["source"], data["run_id"], ModelVersionStatus(data["status"]),

‎src/types/registered_model.jl

+4-4
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ struct RegisteredModelAlias
1111
alias::String
1212
version::String
1313
end
14-
RegisteredModelAlias(data::Dict{String, Any}) = RegisteredModelAlias(data["alias"],
14+
RegisteredModelAlias(data::Dict{String,Any}) = RegisteredModelAlias(data["alias"],
1515
data["version"])
1616
Base.show(io::IO, t::RegisteredModelAlias) = show(io, ShowCase(t, new_lines=true))
1717

@@ -35,13 +35,13 @@ struct RegisteredModel
3535
name::String
3636
creation_timestamp::Int64
3737
last_updated_timestamp::Int64
38-
user_id::Union{String, Nothing}
39-
description::Union{String, Nothing}
38+
user_id::Union{String,Nothing}
39+
description::Union{String,Nothing}
4040
latest_versions::Array{ModelVersion}
4141
tags::Array{Tag}
4242
aliases::Array{RegisteredModelAlias}
4343
end
44-
RegisteredModel(data::Dict{String, Any}) = RegisteredModel(data["name"],
44+
RegisteredModel(data::Dict{String,Any}) = RegisteredModel(data["name"],
4545
data["creation_timestamp"], data["last_updated_timestamp"],
4646
get(data, "user_id", nothing), get(data, "description", nothing),
4747
[ModelVersion(version) for version in get(data, "latest_versions", [])],

0 commit comments

Comments
 (0)