ray.train.get_context

ray.train.get_context() → ray.train.context.TrainContext

Get or create a singleton training context.

The context is only available in a training or tuning loop.

See the TrainContext API reference to see available methods.