Correct typo.

astro-informatics · May 10, 2024 · 892a7f8 · 892a7f8
1 parent a5074ea
commit 892a7f8
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/harmonic/flows.py b/harmonic/flows.py
@@ -264,12 +264,11 @@ def make_flow(self, temperature: float = 1.0):
         flow = distrax.Inverse(distrax.Chain(layers))
 
         if not self.multimodal_base:
-            base_dist = distrax.Independent(
-                distrax.MultivariateNormalFullCovariance(
-                    loc=jnp.zeros(self.n_features),
-                    covariance_matrix=jnp.eye(self.n_features) * temperature,
-                )
+            base_dist = distrax.MultivariateNormalFullCovariance(
+                loc=jnp.zeros(self.n_features),
+                covariance_matrix=jnp.eye(self.n_features) * temperature,
             )
+
         else:
             base_dist = distrax.MixtureOfTwo(
                 0.5,