use only the first class of column

tidyverse · Nov 3, 2024 · 9c74416 · 9c74416
1 parent 3bd3b70
commit 9c74416
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/translate.R b/R/translate.R
@@ -125,7 +125,7 @@ rel_find_call <- function(fun, env) {
 
 infer_class_of_expr <- function(expr, names_data, classes_data) {
   if (typeof(expr) == "symbol" && as.character(expr) %in% names_data) {
-    return(classes_data[which(as.character(expr) == names_data)][[1]])
+    return(classes_data[which(as.character(expr) == names_data)])
   }
   return(class(expr)[[1]])
 }
@@ -358,7 +358,7 @@ rel_translate <- function(
 
   classes_data <- NULL
   if (!missing(data) && !is.null(data)) {
-    classes_data <- map(data, class)
+    classes_data <- unlist(map(data,function(col) class(col)[[1]]))
   }