Fix class check suggestion

adimajo · Mar 5, 2024 · 766bebb · 766bebb
1 parent defb1e5
commit 766bebb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/glmtree.R b/R/glmtree.R
@@ -179,7 +179,7 @@ glmtree <-
       }
 
 
-      if (class(criterion_iter[[i]]) == "matrix") {
+      if (is(criterion_iter[[i]], "matrix")) {
         criterion_iter[[i]] <- normalizedGini(
           actual = criterion_iter[[i]][, 1],
           predicted = criterion_iter[[i]][, 2]