fixed checks

ModelOriented · May 20, 2022 · f41ad09 · f41ad09
1 parent 733b5da
commit f41ad09
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/R/misc_yhat.R b/R/misc_yhat.R
@@ -206,7 +206,7 @@ yhat.train <- function(X.model, newdata, ...) {
 
   }
   # fix for https://github.com/ModelOriented/DALEX/issues/150
-  if (class(response) == "data.frame") response <- as.matrix(response)
+  if (inherits(response, "data.frame")) response <- as.matrix(response)
 
   response
 }

diff --git a/R/model_info.R b/R/model_info.R
@@ -9,7 +9,7 @@
 #' was executed withing \code{explain} function. DALEX will recognize subtype on it's own.
 #' @param ... - another arguments
 #'
-#' @details 
+#' @details
 #' Currently supported packages are:
 #' \itemize{
 #' \item class \code{cv.glmnet} and \code{glmnet} - models created with \pkg{glmnet} package
@@ -284,7 +284,7 @@ print.model_info <- function(x, ...) {
 
 get_pkg_ver_safe <- function(package) {
   ver <- try(as.character(utils::packageVersion(package)), silent = TRUE)
-  if (class(ver) == "try-error") {
+  if (inherits(ver, "try-error")) {
     ver <- "Unknown"
   }
   ver