[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

[kamil-kaczmarek]

This notebook wi tested with ray==2.3.1, have a look at requirements.txt