Use _is_pyarrow_table

microsoft · Jan 9, 2025 · 9ea6d2e · 9ea6d2e
1 parent 2bfe09e
commit 9ea6d2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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