From ddd5fdddcfa5d4a58d8b8c068b7e1e251d0a1278 Mon Sep 17 00:00:00 2001
From: shenkha <legiakhang03@gmail.com>
Date: Wed, 20 Aug 2025 12:23:58 +0400
Subject: [PATCH] Fix api Lib for Ensemble

---
 docs/api/linear.rst          | 2 ++
 libmultilabel/linear/tree.py | 1 +
 2 files changed, 3 insertions(+)

diff --git a/docs/api/linear.rst b/docs/api/linear.rst
index 23361a6b..888db896 100644
--- a/docs/api/linear.rst
+++ b/docs/api/linear.rst
@@ -26,6 +26,8 @@ The simplest usage is::
 
 .. autofunction:: train_tree
 
+.. autofunction:: train_ensemble_tree 
+
 .. autofunction:: predict_values
 
 .. autofunction:: get_topk_labels
diff --git a/libmultilabel/linear/tree.py b/libmultilabel/linear/tree.py
index 34f06370..2e2d83df 100644
--- a/libmultilabel/linear/tree.py
+++ b/libmultilabel/linear/tree.py
@@ -426,6 +426,7 @@ def train_ensemble_tree(
     seed: int = None,
 ) -> EnsembleTreeModel:
     """Trains an ensemble of tree models (Parabel/Bonsai-style).
+    
     Args:
         y (sparse.csr_matrix): A 0/1 matrix with dimensions number of instances * number of classes.
         x (sparse.csr_matrix): A matrix with dimensions number of instances * number of features.