diff --git a/Project.toml b/Project.toml index 9de21e5..4ad6829 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "MLJFlow" uuid = "7b7b8358-b45c-48ea-a8ef-7ca328ad328f" authors = ["Jose Esparza "] -version = "0.3.0" +version = "0.4.0" [deps] MLFlowClient = "64a0f543-368b-4a9a-827a-e71edb2a0b83" diff --git a/src/MLJFlow.jl b/src/MLJFlow.jl index b99d7e5..f42d1b6 100644 --- a/src/MLJFlow.jl +++ b/src/MLJFlow.jl @@ -3,8 +3,8 @@ module MLJFlow using MLJBase: Model, Machine, name using MLJModelInterface: flat_params using MLFlowClient: MLFlow, logparam, logmetric, - createrun, MLFlowRun, updaterun, - healthcheck, logartifact, getorcreateexperiment + createrun, MLFlowRun, updaterun, logartifact, + getorcreateexperiment import Base: show import MLJBase: save, log_evaluation diff --git a/src/types.jl b/src/types.jl index 354bbc1..46a505e 100644 --- a/src/types.jl +++ b/src/types.jl @@ -33,10 +33,6 @@ function Logger(baseuri; experiment_name="MLJ experiment", artifact_location=nothing, verbosity=1) service = MLFlow(baseuri) - if ~healthcheck(service) - error("It seems that the MLFlow server is not running at specified "* - "location, $baseuri. For more information, see https://mlflow.org/docs/latest/quickstart.html") - end Logger(service, verbosity, experiment_name, artifact_location) end function show(io::IO, logger::MLJFlow.Logger) @@ -46,4 +42,4 @@ function show(io::IO, logger::MLJFlow.Logger) " artifact_location=\"$(logger.artifact_location)\",\n" * ") using MLFlow API version $(logger.service.apiversion)" ) -end \ No newline at end of file +end