From 3c01369cb672a552f7d8b58376002957423a6288 Mon Sep 17 00:00:00 2001
From: Jan Ewald <122440612+jan-forest@users.noreply.github.com>
Date: Mon, 20 Jul 2026 10:23:53 +0200
Subject: [PATCH] Remove beta scaling for var_loss in loss DataFrame

Remove scaling of loss_df by config.beta for 'var_loss'.
---
 src/autoencodix/base/_base_visualizer.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/src/autoencodix/base/_base_visualizer.py b/src/autoencodix/base/_base_visualizer.py
index d5a71eec..c7db97c9 100644
--- a/src/autoencodix/base/_base_visualizer.py
+++ b/src/autoencodix/base/_base_visualizer.py
@@ -234,8 +234,6 @@ def _make_loss_format(result: Result, config: DefaultConfig) -> pd.DataFrame:
             loss_df = pd.DataFrame.from_dict(loss_values, orient="index")  # type: ignore
 
             # Rest of your code remains the same
-            if term == "var_loss":
-                loss_df = loss_df * config.beta
             loss_df["Epoch"] = loss_df.index + 1
             loss_df["Loss Term"] = term