data.frame type check

brandmaier · Jul 4, 2024 · 02d6cc4 · 02d6cc4
1 parent 6af4ae7
commit 02d6cc4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/R/semtree.R b/R/semtree.R
@@ -85,6 +85,14 @@
 #' @export
 semtree <- function(model, data = NULL, control = NULL, constraints = NULL,
                     predictors = NULL, ...) {
+
+  # some checks on the data
+  if (!is.null(data)) {
+    if (!is.data.frame(data)) {
+      stop("Error with 'data' argument: semtree currently only supports data frames.")
+    }
+  }
+
   # TODO: change this throughout
   dataset <- data