remove circular reference in DataIter (dmlc#11177)

Co-authored-by: Taewoo Kim <[email protected]>
razdoburdin · Jan 20, 2025 · 7c69d92 · 7c69d92
1 parent 33a3da2
commit 7c69d92
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python-package/xgboost/core.py b/python-package/xgboost/core.py
@@ -508,15 +508,15 @@ def __init__(
     def get_callbacks(self, enable_categorical: bool) -> Tuple[Callable, Callable]:
         """Get callback functions for iterating in C. This is an internal function."""
         assert hasattr(self, "cache_prefix"), "__init__ is not called."
-        self._reset_callback = ctypes.CFUNCTYPE(None, ctypes.c_void_p)(
+        reset_callback = ctypes.CFUNCTYPE(None, ctypes.c_void_p)(
             self._reset_wrapper
         )
-        self._next_callback = ctypes.CFUNCTYPE(
+        next_callback = ctypes.CFUNCTYPE(
             ctypes.c_int,
             ctypes.c_void_p,
         )(self._next_wrapper)
         self._enable_categorical = enable_categorical
-        return self._reset_callback, self._next_callback
+        return reset_callback, next_callback
 
     @property
     def proxy(self) -> "_ProxyDMatrix":