Skip to content

Commit b3b7296

Browse files
Pre-commit fix
1 parent fb84c5c commit b3b7296

4 files changed

Lines changed: 3 additions & 2 deletions

File tree

DashAI/back/models/base_model.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -394,7 +394,6 @@ def compute_metrics(
394394

395395
return results
396396

397-
398397
def prepare_dataset(
399398
self, dataset: "DashAIDataset", is_fit: bool = False
400399
) -> "DashAIDataset":

DashAI/back/optimizers/optuna_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ def report(results, step):
191191
return
192192
trial.report(value, step)
193193
if trial.should_prune():
194-
raise optuna.TrialPruned()
194+
raise optuna.TrialPruned
195195

196196
return report
197197

tests/back/optimizers/test_optuna_pruning_integration.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ def _holdout_evaluate(model, input_dataset, output_dataset, metric):
4242
None, model, input_dataset, output_dataset, metric
4343
)
4444

45+
4546
EPOCHS = 12
4647
N_TRIALS = 10
4748
# MedianPruner never prunes its first `n_startup_trials` (5 by default): with

tests/back/optimizers/test_optuna_real_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@ def _holdout_evaluate(model, input_dataset, output_dataset, metric):
5353
None, model, input_dataset, output_dataset, metric
5454
)
5555

56+
5657
EPOCHS = 3
5758
N_TRIALS = 3
5859

0 commit comments

Comments
 (0)