formatting update

perpetual-ml · Nov 19, 2024 · a938441 · a938441
1 parent 6bf8675
commit a938441
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/tests/test_booster.py b/python-package/tests/test_booster.py
@@ -35,7 +35,7 @@ def test_booster_max_cat(X_y):
     num_cols = X.select_dtypes(include=np.number).columns.tolist()
     all_cols = X.columns.tolist()
     cat_cols = [x for x in all_cols if x not in num_cols]
-    X[cat_cols] = X[cat_cols].astype('category')
+    X[cat_cols] = X[cat_cols].astype("category")
 
     model = PerpetualBooster(objective="LogLoss", max_cat=4)
     model.fit(X, y)