From bb144010d5f3e13c07779bb91f1de921ba6bc415 Mon Sep 17 00:00:00 2001
From: Louis <louis.lacoste@agroparistech.fr>
Date: Thu, 11 Dec 2025 10:32:55 +0100
Subject: [PATCH] Adding plot to no expit version

---
 pyvbmc_optimization_noexpit.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pyvbmc_optimization_noexpit.py b/pyvbmc_optimization_noexpit.py
index 6b54bc5..5d1b765 100644
--- a/pyvbmc_optimization_noexpit.py
+++ b/pyvbmc_optimization_noexpit.py
@@ -551,6 +551,7 @@ def fit_all_participants(
     method: str = "VBMC",
     n_participants: Optional[int] = None,
     verbose: bool = True,
+    plot: bool = True,
 ) -> Dict[str, List[Dict]]:
     """
     Ajuste tous les modèles pour tous les participants.
@@ -591,7 +592,7 @@ def fit_all_participants(
             try:
                 if method == "VBMC":
                     result = fit_participant_pyvbmc(
-                        participant_data, model_config, verbose=False
+                        participant_data, model_config, verbose=False, plot=plot
                     )
                 else:
                     result = fit_participant_deoptim(
@@ -877,6 +878,7 @@ if __name__ == "__main__":
         method=method,
         n_participants=1,  # Set to a number to limit for testing
         verbose=True,
+        plot=True,
     )
 
     # Comparaison des modèles