refactor: migrate recall_tree finish_example (VowpalWabbit#4402)

bassmang · Jan 4, 2023 · 59482fc · 59482fc
1 parent 66d5579
commit 59482fc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/vowpalwabbit/core/src/reductions/recall_tree.cc b/vowpalwabbit/core/src/reductions/recall_tree.cc
@@ -552,7 +552,9 @@ base_learner* VW::reductions::recall_tree_setup(VW::setup_base_i& stack_builder)
   auto* l = make_reduction_learner(std::move(tree), as_singleline(stack_builder.setup_base_learner()), learn, predict,
       stack_builder.get_setupfn_name(recall_tree_setup))
                 .set_params_per_weight(ws)
-                .set_finish_example(VW::details::finish_multiclass_example<recall_tree&>)
+                .set_update_stats(VW::details::update_stats_multiclass_label<recall_tree>)
+                .set_output_example_prediction(VW::details::output_example_prediction_multiclass_label<recall_tree>)
+                .set_print_update(VW::details::print_update_multiclass_label<recall_tree>)
                 .set_save_load(save_load_tree)
                 .set_output_prediction_type(VW::prediction_type_t::MULTICLASS)
                 .set_input_label_type(VW::label_type_t::MULTICLASS)