caikit.runtime.servicers.training_management_servicer

Attributes

Classes

TrainingManagementServicerImpl

This class contains the implementation of all of the RPCs that are required to run a

Module Contents

caikit.runtime.servicers.training_management_servicer.log[source]

class caikit.runtime.servicers.training_management_servicer.TrainingManagementServicerImpl[source]

This class contains the implementation of all of the RPCs that are required to run a service in Model Mesh as a Model-Runtime.

GetTrainingStatus(request, context)[source]: Get the status of a training by ID

CancelTraining(request, context)[source]: Cancel a training future.

get_training_status(training_id: str) → caikit.interfaces.runtime.data_model.TrainingStatusResponse[source]: Get the status of a training by ID

cancel_training(training_id: str) → caikit.interfaces.runtime.data_model.TrainingStatusResponse[source]: Cancel a training future.

static _get_model_future(training_id: str, operation: str)[source]: Returns a model future, or raises 404 caikit runtime exception on error. Wrapped here so that we only catch errors directly in the trainer.get_model_future call