Avoid grouping keys for tsibbles without keys

tidyverts · Sep 16, 2019 · 00d67d9 · 00d67d9
1 parent 6293474
commit 00d67d9
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/R/features.R b/R/features.R
@@ -12,17 +12,12 @@ features_impl <- function(.tbl, .var, features, ...){
 
   # Compute response
   key_dt <- key_data(.tbl)
-  .resp <- unclass(
-    dplyr::transmute(
-      dplyr::new_grouped_df(as_tibble(.tbl), key_dt),
-      !!!.var
-    )
-  )
-
-  if(NCOL(key_dt) > 1){
-    .resp <- .resp[-seq_len(NCOL(key_dt) - 1)]
+  .tbl <- as_tibble(.tbl)
+  if(NROW(key_dt) > 1){
+    .tbl <- dplyr::new_grouped_df(.tbl, key_dt)
   }
-
+  .resp <- unclass(dplyr::transmute(.tbl, !!!.var))
+  .resp <- .resp[seq_along(.var) + NCOL(key_dt) - 1]
   names(.resp) <- names(.var)
 
   # Compute features