From 0e7343ec19dec0fae44ce5f3ef612f47cd9e3fed Mon Sep 17 00:00:00 2001 From: Zhe Zhang Date: Wed, 27 Jan 2021 17:16:29 -0800 Subject: [PATCH] [docs] Fix MLflow / Tune example in documentation (#13740) Minor fixes to make it runnable --- python/ray/tune/integration/mlflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/ray/tune/integration/mlflow.py b/python/ray/tune/integration/mlflow.py index cbd3811d4..6e038b810 100644 --- a/python/ray/tune/integration/mlflow.py +++ b/python/ray/tune/integration/mlflow.py @@ -274,8 +274,8 @@ def mlflow_mixin(func: Callable): @mlflow_mixin def train_fn(config): for i in range(10): - loss = self.config["a"] + self.config["b"] - mlflow.log_metric(key="loss", value=loss}) + loss = config["a"] + config["b"] + mlflow.log_metric(key="loss", value=loss) tune.report(loss=loss, done=True) tune.run(