infer class only returns first class

tidyverse · Nov 2, 2024 · 3bd3b70 · 3bd3b70
1 parent 6310010
commit 3bd3b70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/translate.R b/R/translate.R
@@ -127,7 +127,7 @@ 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(class(expr))
+  return(class(expr)[[1]])
 }
 
 classes_are_comparable <- function(left, right) {