diff --git a/python-package/lightgbm/basic.py b/python-package/lightgbm/basic.py index 3a07cc1cb8f3..1834994e6889 100644 --- a/python-package/lightgbm/basic.py +++ b/python-package/lightgbm/basic.py @@ -1364,7 +1364,7 @@ def __init__( free_raw_data : bool, optional (default=True) If True, raw data is freed after constructing inner Dataset. """ - self.handle: Optional[ctypes.c_void_p] = None + self.handle: Optional[_DatasetHandle] = None self.data = data self.label = label self.reference = reference