pass call to tune::first_eval_time() (#103)

tidymodels · Jan 25, 2024 · b18c9c5 · b18c9c5
1 parent e86f4b0
commit b18c9c5
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 4 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: finetune
 Title: Additional Functions for Model Tuning
-Version: 1.1.0.9005
+Version: 1.1.0.9006
 Authors@R: c(
     person("Max", "Kuhn", , "[email protected]", role = c("aut", "cre"),
            comment = c(ORCID = "0000-0003-2402-136X")),

diff --git a/R/tune_race_anova.R b/R/tune_race_anova.R
@@ -262,7 +262,12 @@ tune_race_anova_workflow <-
     opt_metric_name <- opt_metric$metric
     maximize <- opt_metric$direction == "maximize"
 
-    opt_metric_time <- tune::first_eval_time(metrics, opt_metric_name, eval_time)
+    opt_metric_time <- tune::first_eval_time(
+      metrics,
+      opt_metric_name,
+      eval_time,
+      call = call
+    )
 
     racing_obj_log(opt_metric_name, opt_metric$direction, control, opt_metric_time)
 

diff --git a/R/tune_race_win_loss.R b/R/tune_race_win_loss.R
@@ -246,7 +246,12 @@ tune_race_win_loss_workflow <-
     opt_metric_name <- opt_metric$metric
     maximize <- opt_metric$direction == "maximize"
 
-    opt_metric_time <- tune::first_eval_time(metrics, opt_metric_name, eval_time)
+    opt_metric_time <- tune::first_eval_time(
+      metrics,
+      opt_metric_name,
+      eval_time,
+      call = call
+    )
 
     racing_obj_log(opt_metric_name, opt_metric$direction, control, opt_metric_time)
 

diff --git a/R/tune_sim_anneal.R b/R/tune_sim_anneal.R
@@ -321,7 +321,12 @@ tune_sim_anneal_workflow <-
     maximize <- opt_metric$direction == "maximize"
 
     eval_time <- tune::check_eval_time_arg(eval_time, metrics, call = call)
-    opt_metric_time <- tune::first_eval_time(metrics, opt_metric_name, eval_time)
+    opt_metric_time <- tune::first_eval_time(
+        metrics,
+        opt_metric_name,
+        eval_time,
+        call = call
+      )
 
     if (is.null(param_info)) {
       param_info <- extract_parameter_set_dials(object)