Ensure predict methods are using process_x utils

davidrsch · davidrsch · commit a206d80528bb · 2025-08-13T01:12:34.000+02:00
diff --git a/R/register_fit_predict.R b/R/register_fit_predict.R
@@ -57,7 +57,7 @@ register_fit_predict <- function(model_name, mode, layer_blocks, functional) {
         func = c(fun = "predict"),
         args = list(
           object = rlang::expr(object$fit$fit),
-          x = rlang::expr(as.matrix(new_data))
+          x = rlang::expr(process_x(new_data)$x_proc)
         )
       )
     )
@@ -74,7 +74,7 @@ register_fit_predict <- function(model_name, mode, layer_blocks, functional) {
         func = c(fun = "predict"),
         args = list(
           object = rlang::expr(object$fit$fit),
-          x = rlang::expr(as.matrix(new_data))
+          x = rlang::expr(process_x(new_data)$x_proc)
         )
       )
     )
@@ -89,7 +89,7 @@ register_fit_predict <- function(model_name, mode, layer_blocks, functional) {
         func = c(fun = "predict"),
         args = list(
           object = rlang::expr(object$fit$fit),
-          x = rlang::expr(as.matrix(new_data))
+          x = rlang::expr(process_x(new_data)$x_proc)
         )
       )
     )

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ register_fit_predict <- function(model_name, mode, layer_blocks, functional) {`
`57`	`57`	`func = c(fun = "predict"),`
`58`	`58`	`args = list(`
`59`	`59`	`object = rlang::expr(object$fit$fit),`
`60`		`- x = rlang::expr(as.matrix(new_data))`
	`60`	`+ x = rlang::expr(process_x(new_data)$x_proc)`
`61`	`61`	`)`
`62`	`62`	`)`
`63`	`63`	`)`
`@@ -74,7 +74,7 @@ register_fit_predict <- function(model_name, mode, layer_blocks, functional) {`
`74`	`74`	`func = c(fun = "predict"),`
`75`	`75`	`args = list(`
`76`	`76`	`object = rlang::expr(object$fit$fit),`
`77`		`- x = rlang::expr(as.matrix(new_data))`
	`77`	`+ x = rlang::expr(process_x(new_data)$x_proc)`
`78`	`78`	`)`
`79`	`79`	`)`
`80`	`80`	`)`
`@@ -89,7 +89,7 @@ register_fit_predict <- function(model_name, mode, layer_blocks, functional) {`
`89`	`89`	`func = c(fun = "predict"),`
`90`	`90`	`args = list(`
`91`	`91`	`object = rlang::expr(object$fit$fit),`
`92`		`- x = rlang::expr(as.matrix(new_data))`
	`92`	`+ x = rlang::expr(process_x(new_data)$x_proc)`
`93`	`93`	`)`
`94`	`94`	`)`
`95`	`95`	`)`