fix: unneeded func params

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):

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()