caikit.runtime

Submodules

Functions

get_inference_request(...)

Helper function to return the inference request DataModel for the Module or Task Class

get_train_params(...)

Helper function to return the train parameters DataModel for the Module Class

get_train_request(...)

Helper function to return the train request DataModel for the Module Class

Package Contents

caikit.runtime.get_inference_request(task_or_module_class: Type[caikit.core.ModuleBase | caikit.core.task.TaskBase], input_streaming: bool = False, output_streaming: bool = False) Type[caikit.core.data_model.base.DataBase][source]

Helper function to return the inference request DataModel for the Module or Task Class

caikit.runtime.get_train_params(module_class: Type[caikit.core.ModuleBase]) Type[caikit.core.data_model.base.DataBase][source]

Helper function to return the train parameters DataModel for the Module Class

caikit.runtime.get_train_request(module_class: Type[caikit.core.ModuleBase]) Type[caikit.core.data_model.base.DataBase][source]

Helper function to return the train request DataModel for the Module Class