fix: [<-, [[<-, $<- on grouped epi_dfs: drop edf-ness only when necessary

Author: brookslogan

NAMESPACE

@@ -1,6 +1,9 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method("$<-",epi_df)
 S3method("[",epi_df)
+S3method("[<-",epi_df)
+S3method("[[<-",epi_df)
 S3method("names<-",epi_df)
 S3method(Summary,epi_df)
 S3method(arrange_canonical,default)

R/methods-epi_df.R

@@ -235,6 +235,27 @@ reconstruct_light_edf <- function(data, template) {
   reconstruct_light_edf(res, x)
 }
 
+#' @export
+`[<-.epi_df` <- function(x, ...) {
+  res <- NextMethod()
+
+  reconstruct_light_edf(res, x)
+}
+
+#' @export
+`[[<-.epi_df` <- function(x, ...) {
+  res <- NextMethod()
+
+  reconstruct_light_edf(res, x)
+}
+
+#' @export
+`$<-.epi_df` <- function(x, ...) {
+  res <- NextMethod()
+
+  reconstruct_light_edf(res, x)
+}
+
 #' @importFrom dplyr dplyr_col_modify
 #' @export
 dplyr_col_modify.epi_df <- function(data, cols) {