From b76c5d54bf4ba153f03bf73f6c9aa5ee034df659 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Sat, 19 Feb 2022 18:52:41 +0800
Subject: [PATCH] Define export symbols in callback module. (#7665)

---
 python-package/xgboost/callback.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/python-package/xgboost/callback.py b/python-package/xgboost/callback.py
index 901724a67..b44b92c02 100644
--- 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]]]