Bugfix/int serde (#37)

kevinykuo · web-flow · commit 54786046de1d · 2019-11-05T19:20:41.000-08:00
* fix integer serialization

* update news
diff --git a/NEWS.md b/NEWS.md
@@ -2,6 +2,7 @@
 
 * **Breaking change**: `ml_write_bundle()` now takes the `sample_input` instead of `dataset`.
 * Support for Spark 2.4 and MLeap 0.14.
+* Integer inputs are now supported (#36).
 
 # mleap 0.1.3
 
diff --git a/R/prediction.R b/R/prediction.R
@@ -16,6 +16,8 @@ mleap_transform <- function(model, data) {
   types <- columns %>%
     purrr::map_chr(~ input_schema$type[[match(.x, input_schema$name)]])
   
+  types <- gsub("^int$", "integer", types)
+  
   schema <- list(fields = purrr::map2(
     columns, types, ~ list(name = .x, type = .y))
   )

Original file line number	Diff line number	Diff line change
`@@ -16,6 +16,8 @@ mleap_transform <- function(model, data) {`
`16`	`16`	`types <- columns %>%`
`17`	`17`	`purrr::map_chr(~ input_schema$type[[match(.x, input_schema$name)]])`
`18`	`18`
	`19`	`+ types <- gsub("^int$", "integer", types)`
	`20`	`+`
`19`	`21`	`schema <- list(fields = purrr::map2(`
`20`	`22`	`columns, types, ~ list(name = .x, type = .y))`
`21`	`23`	`)`