system.time(); prettyunis

Author: patr1ckm

DESCRIPTION

@@ -28,6 +28,7 @@ Imports:
     stats,
     tidyr,
     globals,
+    prettyunits,
     vctrs (>= 0.2.0)
 Roxygen: list(markdown = TRUE)
 RoxygenNote: 6.1.99.9001

R/fit.R

@@ -367,8 +367,7 @@ check_xy_interface <- function(x, y, cl, model) {
 #' @export
 print.model_fit <- function(x, ...) {
   cat("parsnip model object\n\n")
-  cat("Fit in: ", paste(format(unclass(x$elapsed), digits = 2),
-                        attr(x$elapsed, "units")))
+  cat("Fit in: ", prettyunits::pretty_sec(x$elapsed[["elapsed"]]))
 
   if (inherits(x$fit, "try-error")) {
     cat("Model fit failed with error:\n", x$fit, "\n")

R/fit_helpers.R

@@ -52,19 +52,17 @@ form_form <-
       spec = object
     )
 
-    start <- Sys.time()
-
-    res$fit <- eval_mod(
-      fit_call,
-      capture = control$verbosity == 0,
-      catch = control$catch,
-      env = env,
-      ...
+    elapsed <- system.time(
+      res$fit <- eval_mod(
+        fit_call,
+        capture = control$verbosity == 0,
+        catch = control$catch,
+        env = env,
+        ...
+      )
     )
-
-    end <- Sys.time()
     res$preproc <- list(y_var = all.vars(env$formula[[2]]))
-    res$elapsed <- end - start
+    res$elapsed <- elapsed
     res
   }
 
@@ -112,25 +110,24 @@ xy_xy <- function(object, env, control, target = "none", ...) {
 
   res <- list(lvl = levels(env$y), spec = object)
 
-  start <- Sys.time()
 
-  res$fit <- eval_mod(
-    fit_call,
-    capture = control$verbosity == 0,
-    catch = control$catch,
-    env = env,
-    ...
+  elapsed <- system.time(
+    res$fit <- eval_mod(
+      fit_call,
+      capture = control$verbosity == 0,
+      catch = control$catch,
+      env = env,
+      ...
+    )
   )
 
-  end <- Sys.time()
-
   if (is.vector(env$y)) {
     y_name <- character(0)
   } else {
     y_name <- colnames(env$y)
   }
   res$preproc <- list(y_var = y_name)
-  res$elapsed <- end - start
+  res$elapsed <- elapsed
   res
 }