[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
pyg-team · Sep 18, 2024 · 830ce79 · 830ce79
1 parent e49c9f0
commit 830ce79
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/torch_frame/gbdt/tuned_lightgbm.py b/torch_frame/gbdt/tuned_lightgbm.py
@@ -166,8 +166,7 @@ def objective(
 
         boost = lightgbm.train(
             self.params, train_data, num_boost_round=num_boost_round,
-            valid_sets=[eval_data],
-            callbacks=[
+            valid_sets=[eval_data], callbacks=[
                 lightgbm.early_stopping(stopping_rounds=50, verbose=False),
                 lightgbm.log_evaluation(period=2000)
             ])
@@ -202,14 +201,12 @@ def _tune(
 
         study.optimize(
             lambda trial: self.objective(trial, train_data, eval_data,
-                                         num_boost_round),
-            num_trials)
+                                         num_boost_round), num_trials)
         self.params.update(study.best_params)
 
         self.model = lightgbm.train(
             self.params, train_data, num_boost_round=num_boost_round,
-            valid_sets=[eval_data],
-            callbacks=[
+            valid_sets=[eval_data], callbacks=[
                 lightgbm.early_stopping(stopping_rounds=50, verbose=False),
                 lightgbm.log_evaluation(period=2000)
             ])