use only the first class of column

tidyverse · Nov 6, 2024 · e8a5627 · e8a5627
1 parent a9bfa59
commit e8a5627
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]])
 }
@@ -362,7 +362,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]]))
   }