Wrong Input type in onnx_utils.py file

[arushij1711]

def convert_to_onnx(model, data): initial_type = [('float_input', FloatTensorType([None, data.shape[0]]))] onnx_model = convert_sklearn(model, initial_types=initial_type) print("onnx_model.type:",type(onnx_model)) mlflow.set_tag("onnx_version",onnx.version) return onnx_model

It should be data.shape[1]