ray.train.get_context

ray.train.get_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.