PUBDEV-2150 Imports time columns from H2O as Date columns in R.

h2oai · Sep 26, 2015 · 42c2bb4 · jangorecki · Nov 17, 2016 · 42c2bb4
1 parent 68e897e
commit 42c2bb4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/h2o-r/h2o-package/R/frame.R b/h2o-r/h2o-package/R/frame.R
@@ -1730,7 +1730,10 @@ as.data.frame.Frame <- function(x, ...) {
   # Substitute NAs for blank cells rather than skipping
   df <- read.csv((tcon <- textConnection(ttt)), blank.lines.skip = FALSE, na.strings = "", colClasses = colClasses, ...)
   close(tcon)
-  # FIXME now convert all date columns to POSIXct
+  # Convert all date columns to POSIXct
+  dates <- attr(x, "types") %in% "time"
+  if (length(dates) > 0) # why do some frames come in with no attributes but many columns?
+    for (i in 1:length(dates)) { if (dates[[i]]) class(df[[i]]) = "POSIXct" }
   df
 }