kumoai.trainer.export_model

Contents

kumoai.trainer.export_model#

kumoai.trainer.export_model(config, *, non_blocking=True)[source]#

Export model files and batch prediction embeddings to an external storage location for use in online serving.

The export copies online serving model dir in its entirety to the output path, along with the embeddings.parquet result from the BP job specified.

Parameters:

  • config (ModelOutputConfig) – A ModelOutputConfig specifying the training job, output path, and batch prediction job to bundle.

  • non_blocking (bool) – If True, returns an ArtifactExportJob future. If False, blocks until complete and returns an ArtifactExportResult.

Return type:

Union[ArtifactExportJob, ArtifactExportResult]

Contents