From fedb9da31e1e2eaa82d91bc8a462cc1e36c36b0e Mon Sep 17 00:00:00 2001
From: Mehdi Seifi <mehdiseifi@gmail.com>
Date: Thu, 16 Jan 2025 12:27:45 +0100
Subject: [PATCH] fixed clear plot on training reset

---
 src/careamics_napari/widgets/train_progress_widget.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/careamics_napari/widgets/train_progress_widget.py b/src/careamics_napari/widgets/train_progress_widget.py
index 7483317..d2a6c1e 100644
--- a/src/careamics_napari/widgets/train_progress_widget.py
+++ b/src/careamics_napari/widgets/train_progress_widget.py
@@ -91,7 +91,7 @@ def _update_training_state(self: Self, state: TrainingState) -> None:
             Training state.
         """
         if state == TrainingState.IDLE or state == TrainingState.TRAINING:
-            self.plot.clear()
+            self.plot.clear_plot()
 
     def _update_max_epoch(self: Self, max_epoch: int):
         """Update the maximum number of epochs in the progress bar.