aggregate_impute() handles cases of aggregatedImputed with joins lead…

…ing to empty sets.
inbo · Aug 10, 2023 · 5fed6b2 · 5fed6b2
1 parent 6ca5dc7
commit 5fed6b2
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/R/aggregate_impute.R b/R/aggregate_impute.R
@@ -209,6 +209,27 @@ setMethod(
       }
     }
 
+    if (nrow(data) == 0) {
+      return(
+        new(
+          "aggregatedImputed",
+          Covariate = data |>
+            select(!!!grouping) |>
+            as.data.frame(),
+          Imputation = ncol(imputation) |>
+            seq_len() |>
+            sprintf(fmt = "Imputation%04i") |>
+            list() |>
+            c(list(character(0))) |>
+            rev() |>
+            matrix(
+              data = NA_real_, nrow = 0, ncol = ncol(imputation),
+              byrow = FALSE
+            )
+        )
+      )
+    }
+
     data |>
       inner_join(imputation, by = id_column) |>
       group_by(!!!grouping) |>