Export the ensemble model to deploy on triton

export_ensemble.py

from merlin.systems.dag.ops.workflow import TransformWorkflow
from merlin.systems.dag.ensemble import Ensemble
import nvtabular as nvt
import os


BASE_DIR = "."
if not os.path.isdir(os.path.join(BASE_DIR, 'ensemble')):
    os.makedirs(os.path.join(BASE_DIR, 'ensemble'))
    
saved_workflow = nvt.Workflow.load('./data/simulated/processed_dataset')
user_graph = saved_workflow.get_subworkflow('user')

retrieval = ["col1", "col2"] >> TransformWorkflow(user_graph)

# define the path where all the models and config files exported to
export_path = os.path.join(BASE_DIR, 'ensemble')

ensemble = Ensemble(retrieval, user_graph.input_schema)
ens_config, node_configs = ensemble.export(export_path)