Skip to content

Commit d9b1a15

Browse files
authored
minor bug fix (#551)
It is not possible to access params through function signiture, the key should be metric_str
1 parent 7970abf commit d9b1a15

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/pytorch_tabular/tabular_model_tuner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -409,7 +409,7 @@ def tune(
409409
params.update({"trial_id": i})
410410
trials.append(params)
411411
if verbose:
412-
logger.info(f"Trial {i+1}/{n_trials}: {params} | Score: {params[metric]}")
412+
logger.info(f"Trial {i+1}/{n_trials}: {params} | Score: {params[metric_str]}")
413413

414414
trials_df = pd.DataFrame(trials)
415415
trials = trials_df.pop("trial_id")

0 commit comments

Comments
 (0)