put branch back

microsoft · May 9, 2023 · 51ab03a · 51ab03a
1 parent 7922149
commit 51ab03a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python-package/lightgbm/basic.py b/python-package/lightgbm/basic.py
@@ -690,6 +690,8 @@ def _data_from_pandas(
                 feature_name = list(data.columns)
             if categorical_feature == 'auto':  # use cat cols from DataFrame
                 categorical_feature = cat_cols_not_ordered
+            else:  # use cat cols specified by user
+                categorical_feature = list(categorical_feature)
         if feature_name == 'auto':
             feature_name = list(data.columns)
         _check_for_bad_pandas_dtypes(data.dtypes)