merlin.systems.triton.convert_df_to_triton_input

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]

Contents