From 0ebc106fd52ce59d38a0f5847178919ccb0d3f0d Mon Sep 17 00:00:00 2001
From: "Guillaume \"Vermeille\" Sanchez" <guillaume.sanchez@lightonai.com>
Date: Wed, 4 Sep 2024 13:56:40 +0000
Subject: [PATCH] hotfix

---
 torchelie/nn/layers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchelie/nn/layers.py b/torchelie/nn/layers.py
index 9e6075d..8219660 100644
--- a/torchelie/nn/layers.py
+++ b/torchelie/nn/layers.py
@@ -224,7 +224,7 @@ def forward(self, x):
             out = tchf.unit_gaussian_prior(mu, sigma, strength)
             return out if not self.return_std else (out, sigma)
         else:
-            return mu if not self.return_std else (mu, sigma
+            return mu if not self.return_std else (mu, sigma)
 
 
 class InformationBottleneck(UnitGaussianPrior):