Update doclayout_yolo/utils/callbacks/raytune.py

doclayout_yolo/utils/callbacks/raytune.py

@@ -7,19 +7,17 @@ try:
     import ray
     from ray import tune
     from ray.air import session
-
 except (ImportError, AssertionError):
     tune = None
-
+    session = None
 
 def on_fit_epoch_end(trainer):
     """Sends training metrics to Ray Tune at end of each epoch."""
-    if ray.tune.is_session_enabled():
+    if session and session.get_session():  # ✅ this is the correct Ray way now
         metrics = trainer.metrics
         metrics["epoch"] = trainer.epoch
         session.report(metrics)
 
-
 callbacks = (
     {
         "on_fit_epoch_end": on_fit_epoch_end,