From 10e170c490cb7be64c731ee1149a30ab0f439df1 Mon Sep 17 00:00:00 2001
From: Dominique Makowski <dom.mak19@gmail.com>
Date: Tue, 6 Jul 2021 16:15:49 +0800
Subject: [PATCH] minor

---
 R/estimate_predicted.R | 2 +-
 README.Rmd             | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/R/estimate_predicted.R b/R/estimate_predicted.R
index b50a3e0df..3e3fa466a 100644
--- 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
index 404cbb99e..8686e57b2 100644
--- 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()
 ```