fix one more issue

sdpython · Jul 4, 2023 · 05f2fd0 · 05f2fd0
1 parent a61d63b
commit 05f2fd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnx_extended/ortops/optim/cpu/tree_ensemble.h b/onnx_extended/ortops/optim/cpu/tree_ensemble.h
@@ -10,7 +10,7 @@ struct TreeEnsembleKernel {
   void Compute(OrtKernelContext* context);
 
   // Attributes
-  onnx_c_ops::RuntimeTreeEnsembleCommon<float>* reg_float;
+  onnx_c_ops::TreeEnsembleCommon<float, float, float>* reg_float;
 };
 
 struct TreeEnsembleRegressor : Ort::CustomOpBase<TreeEnsembleRegressor, TreeEnsembleKernel> {