Doc fix

dmlc · Oct 25, 2023 · 82b48c6 · 82b48c6
1 parent 31782c1
commit 82b48c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/treelite/sklearn/importer.py b/python/treelite/sklearn/importer.py
@@ -101,8 +101,8 @@ def import_model(sklearn_model):
       import treelite.sklearn
       model = treelite.sklearn.import_model(clf)
 
-    Notes
-    -----
+    Note
+    ----
     This function does not yet support categorical splits in HistGradientBoostingRegressor and
     HistGradientBoostingClassifier. If you are using either estimator types, make sure that all
     test nodes have numerical test conditions.