From f2d25f62bbb1d431e9ca139835c579c25636853f Mon Sep 17 00:00:00 2001
From: angelinalg <122562471+angelinalg@users.noreply.github.com>
Date: Thu, 20 Apr 2023 20:36:11 -0700
Subject: [PATCH] [docs] replace tune.report with session.report (#34435)

---
 python/ray/tune/tests/tutorial.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/ray/tune/tests/tutorial.py b/python/ray/tune/tests/tutorial.py
index ceb48366334f0..7016ff18be92e 100644
--- a/python/ray/tune/tests/tutorial.py
+++ b/python/ray/tune/tests/tutorial.py
@@ -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
@@ -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