remove spark bits

EmilHvitfeldt · EmilHvitfeldt · commit 93926ebc8c53 · 2020-12-03T00:02:17.000-08:00
diff --git a/R/predict_linear_pred.R b/R/predict_linear_pred.R
@@ -32,10 +32,8 @@ predict_linear_pred.model_fit <- function(object, new_data, ...) {
 
   if (is.vector(res)) {
     res <- unname(res)
-  } else {
-    if (!inherits(res, "tbl_spark"))
-      res <- as.data.frame(res)
   }
+
   res
 }
 
diff --git a/R/predict_time.R b/R/predict_time.R
@@ -36,10 +36,8 @@ predict_time.model_fit <- function(object, new_data, ...) {
 
   if (is.vector(res)) {
     res <- unname(res)
-  } else {
-    if (!inherits(res, "tbl_spark"))
-      res <- as.data.frame(res)
   }
+
   res
 }
 

Original file line number	Diff line number	Diff line change
`@@ -32,10 +32,8 @@ predict_linear_pred.model_fit <- function(object, new_data, ...) {`
`32`	`32`
`33`	`33`	`if (is.vector(res)) {`
`34`	`34`	`res <- unname(res)`
`35`		`- } else {`
`36`		`- if (!inherits(res, "tbl_spark"))`
`37`		`- res <- as.data.frame(res)`
`38`	`35`	`}`
	`36`	`+`
`39`	`37`	`res`
`40`	`38`	`}`
`41`	`39`
Original file line number	Diff line number	Diff line change
`@@ -36,10 +36,8 @@ predict_time.model_fit <- function(object, new_data, ...) {`
`36`	`36`
`37`	`37`	`if (is.vector(res)) {`
`38`	`38`	`res <- unname(res)`
`39`		`- } else {`
`40`		`- if (!inherits(res, "tbl_spark"))`
`41`		`- res <- as.data.frame(res)`
`42`	`39`	`}`
	`40`	`+`
`43`	`41`	`res`
`44`	`42`	`}`
`45`	`43`