[train] fix job working dir test (#58010)

cherrypick #57995

Signed-off-by: Lonnie Liu <[email protected]>

Author: aslonnie

release/jobs_tests/workloads/run_simple_tune_job.py

@@ -1,7 +1,7 @@
 # From https://docs.ray.io/en/latest/tune/index.html
 
 import ray
-from ray import train, tune
+from ray import tune
 
 
 def objective(step, alpha, beta):
@@ -15,7 +15,7 @@ def training_function(config):
         # Iterative training function - can be any arbitrary training procedure.
         intermediate_score = objective(step, alpha, beta)
         # Feed the score back back to Tune.
-        train.report(dict(mean_loss=intermediate_score))
+        tune.report(dict(mean_loss=intermediate_score))
 
 
 ray.init(address="auto")