From adc209df87a65e546c4fd344ff57c27c0d827a6e Mon Sep 17 00:00:00 2001
From: Lorenzo Fabbri <lorenzo.fabbri@isglobal.org>
Date: Thu, 22 Jun 2023 16:24:03 +0200
Subject: [PATCH] Tidying output balance exploration.

---
 R/weighting.R | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/R/weighting.R b/R/weighting.R
index 545c657..a2c9961 100644
--- a/R/weighting.R
+++ b/R/weighting.R
@@ -70,8 +70,7 @@ estimate_weights <- function(dat,
 #' @export
 explore_balance <- function(exposure,
                             covariates,
-                            weights,
-                            path_save) {
+                            weights) {
   threshold_cor <- 0.1
 
   # Assessing balance numerically
@@ -94,9 +93,11 @@ explore_balance <- function(exposure,
                             abs = TRUE,
                             var.order = "unadjusted",
                             thresholds = c(cor = threshold_cor),
-                            line = TRUE)
+                            line = TRUE,
+                            title = exposure)
 
   return(list(
+    exposure = exposure,
     tab = tab,
     graph = graph,
     love = love