merlin.systems.triton.convert_df_to_triton_input#
- merlin.systems.triton.convert_df_to_triton_input(schema, batch, input_class=<class 'tritonclient.grpc._infer_input.InferInput'>, dtype='int32')[source]#
Convert a dataframe to a set of Triton inputs
- Parameters:
schema (Schema) – Schema of the input data
batch (DataFrame) – The input data itself
input_class (Triton input class, optional) – The Triton input class to use, by default grpcclient.InferInput
dtype (str, optional) – The dtype for lengths/offsets values, by default “int32”
- Returns:
A list of Triton inputs of the requested input class
- Return type:
List[input_class]