From 849d988de5833712f5caeba48881cc6001a97e69 Mon Sep 17 00:00:00 2001
From: Lenoctambule <106790775+lenoctambule@users.noreply.github.com>
Date: Fri, 10 Apr 2026 15:00:04 +0200
Subject: [PATCH] feat(autoencoder.py): __str__ method for
 VariationalAutoencoder class

---
 src/easyvae/autoencoder.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/easyvae/autoencoder.py b/src/easyvae/autoencoder.py
index f6f9566..d913699 100644
--- a/src/easyvae/autoencoder.py
+++ b/src/easyvae/autoencoder.py
@@ -96,7 +96,7 @@ class ClassicalAutoencoder(AAutoencoder):
                 f"Type: {__class__.__name__}",
                 "Encoder:",
                 f"{self.encoder}",
-                "Decoder:"
+                "Decoder:",
                 f"{self.decoder}"
             ))
 
@@ -133,6 +133,15 @@ class VariationalAutoencoder(AAutoencoder):
         super().__init__(*args, **kwargs)
         self.sampler = SampleLayer(self.encoder.out_size, self.lr, Identity())
 
+    def __str__(self):
+        return "\n".join((
+                f"Type: {__class__.__name__}",
+                "Encoder:",
+                f"{self.encoder}",
+                "Decoder:",
+                f"{self.decoder}"
+            ))
+
     def loss(self, data_set: list[np.ndarray]) -> float:
         loss = 0
         for x in data_set: