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