caikit.runtime.servicers.model_train_servicer

Attributes

log

Classes

ModelTrainServicerImpl

This class contains the implementation of the Model Train process.proto

Module Contents

caikit.runtime.servicers.model_train_servicer.log

class caikit.runtime.servicers.model_train_servicer.ModelTrainServicerImpl(training_service: caikit.runtime.service_factory.ServicePackage)

Bases: caikit.runtime.protobufs.process_pb2_grpc.ProcessServicer

This class contains the implementation of the Model Train process.proto interface (a.k.a. the Process Servicer).

_training_service

_gts

Run(request, context)

Run RPC – launches a training job. Args:

request(process_pb2.ProcessRequest):

Generated from process.proto

context(grpc.ServicerContext): Context object (contains request metadata, etc)

Returns:

process_pb2.ProcessResponse:

Generated from process.proto

_update_file_references(train_message_request: google.protobuf.message.Message, training_input_dir: str)

static _update_file_ref(file_path: str, training_input_dir: str) → str

Append the training_input_dir to the file_path if the resulting path exists