Skip to content

Commit

Permalink
[docs] replace tune.report with session.report (#34435)
Browse files Browse the repository at this point in the history
  • Loading branch information
angelinalg committed Apr 21, 2023
1 parent 77267a3 commit 24cbd38
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python/ray/tune/tests/tutorial.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
import torch.nn.functional as F

from ray import air, tune
from ray.air import session
from ray.tune.schedulers import ASHAScheduler
# __tutorial_imports_end__
# fmt: on
Expand Down Expand Up @@ -105,7 +106,7 @@ def train_mnist(config):
acc = test(model, test_loader)

# Send the current training result back to Tune
tune.report(mean_accuracy=acc)
session.report(mean_accuracy=acc)

if i % 5 == 0:
# This saves the model to the trial directory
Expand Down

0 comments on commit 24cbd38

Please sign in to comment.