Define export symbols in callback module. (#7665)

2022-02-19 18:52:41 +08:00
parent 7366d3b20c
commit b76c5d54bf
1 changed files with 8 additions and 0 deletions
--- a/python-package/xgboost/callback.py
+++ b/python-package/xgboost/callback.py
@@ -19,6 +19,14 @@ from .core import Booster, DMatrix, XGBoostError, _get_booster_layer_trees
 from .compat import STRING_TYPES


+__all__ = [
+    "TrainingCallback",
+    "LearningRateScheduler",
+    "EarlyStopping",
+    "EvaluationMonitor",
+    "TrainingCheckPoint",
+]
+
 _Score = Union[float, Tuple[float, float]]
 _ScoreList = Union[List[float], List[Tuple[float, float]]]