diff --git a/python/ray/tune/tests/tutorial.py b/python/ray/tune/tests/tutorial.py index 7016ff18be92e..d3b217ca7f8f2 100644 --- a/python/ray/tune/tests/tutorial.py +++ b/python/ray/tune/tests/tutorial.py @@ -106,7 +106,7 @@ def train_mnist(config): acc = test(model, test_loader) # Send the current training result back to Tune - session.report(mean_accuracy=acc) + session.report({"mean_accuracy": acc}) if i % 5 == 0: # This saves the model to the trial directory