Merge branch 'lasso' of https://github.com/FabricioArendTorres/enflows …

…into lasso
FabricioArendTorres · Aug 5, 2024 · 99bdd17 · 99bdd17
2 parents e3f2733 + a35041f
commit 99bdd17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/enflows/distributions/uniform.py b/enflows/distributions/uniform.py
@@ -250,8 +250,8 @@ def __compute_log_surface(self):
         # side_length = np.sqrt(2.)
         # log_num = self.cart_dim * np.log(side_length) + 0.5 * np.log(self.cart_dim)
         # log_den = 2 * sp.special.gammaln(self.cart_dim)
-        log_const =  - sp.special.loggamma(self.cart_dim)
-        all_quadrants = np.log(2) * d if self.extend_star_like else 0.0
+        log_const = - sp.special.loggamma(self.cart_dim)
+        all_quadrants = np.log(2) * self.cart_dim if self.extend_star_like else 0.0
 
         return log_const + all_quadrants