minor

easystats · Jul 6, 2021 · 10e170c · 10e170c
1 parent f4c29c6
commit 10e170c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/R/estimate_predicted.R b/R/estimate_predicted.R
@@ -136,7 +136,7 @@ estimate_response <- estimate_prediction
 
 
   # If a visualisation_matrix is passed
-  if (inherits(model, "visualisation_matrix")) {
+  if (inherits(model, "visualisation_matrix") || all(class(model) == "data.frame")) {
     data_original <- data
     data <- model
     if ("model" %in% names(attributes(model))) {

diff --git a/README.Rmd b/README.Rmd
@@ -123,7 +123,7 @@ ggplot(iris, aes(x = Petal.Length, y = Sepal.Length)) +
   geom_line(data = vizdata, aes(y = Predicted, color = Petal.Width), size = 1) +
   # Improve colors / themes
   scale_color_viridis_d(direction = -1) +
-  scale_fill_viridis_c(guide = FALSE) +
+  scale_fill_viridis_c(guide = "none") +
   theme_modern()
 ```