From b4963513cc83b565ed2e6a9307b12d8d1ec92fa9 Mon Sep 17 00:00:00 2001
From: Lenoctambule <106790775+lenoctambule@users.noreply.github.com>
Date: Thu, 9 Apr 2026 19:19:59 +0200
Subject: [PATCH] fix: unneeded func params

---
 src/easyvae/autoencoder.py | 2 +-
 src/easyvae/utils.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/easyvae/autoencoder.py b/src/easyvae/autoencoder.py
index e4a9be9..5f436c9 100644
--- a/src/easyvae/autoencoder.py
+++ b/src/easyvae/autoencoder.py
@@ -64,7 +64,7 @@ class AAutoencoder(ABC):
                     break
                 epoch += 1
         if display_loss is True:
-            dynamic_loss_plot_finish(ax, line)
+            dynamic_loss_plot_finish()
         return losses
 
     def save(self, path: str):
diff --git a/src/easyvae/utils.py b/src/easyvae/utils.py
index e407bff..7e61971 100644
--- a/src/easyvae/utils.py
+++ b/src/easyvae/utils.py
@@ -41,6 +41,6 @@ def dynamic_loss_plot_update(ax, line, loss):
     plt.pause(0.1)
 
 
-def dynamic_loss_plot_finish(ax, line):
+def dynamic_loss_plot_finish():
     plt.ioff()
     plt.show()