[Bug]: ray.air checkpoints has moved to ray.train checkpoints

[ajikmr]

Notebook with bug

Computer_vision_workloads/Semantic_segmentation/Scaling_batch_inference.ipynb

What happened?

Import as well as other dependencies need to be fixed for chekpoint related changes.

#from ray.air import Checkpoint from ray.train import Checkpoint

Futher Checkpoint.from_dict() does not work as:

AttributeError: The new ray.train.Checkpoint class does not support from_dict(). Instead, only directories are supported.

Environment info

Ray 2.10.0 Python 3.10.13 Ubuntu

Issue Severity

None