From 9ea6d2e336b260cc1b63a2adc50c70c5d17c64bc Mon Sep 17 00:00:00 2001 From: Malte Londschien Date: Thu, 9 Jan 2025 21:18:47 +0100 Subject: [PATCH] Use _is_pyarrow_table --- python-package/lightgbm/basic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/lightgbm/basic.py b/python-package/lightgbm/basic.py index caa11b39f6d8..36b6baf55ddf 100644 --- a/python-package/lightgbm/basic.py +++ b/python-package/lightgbm/basic.py @@ -2126,7 +2126,7 @@ def _lazy_init( categorical_feature=categorical_feature, pandas_categorical=self.pandas_categorical, ) - elif isinstance(data, pa_Table): + elif _is_pyarrow_table(data): feature_name = data.column_names # process for args