Update conditional_mixture_gaussian_node.py

aimclub · Oct 11, 2023 · 8a862a2 · 8a862a2
1 parent bf1b476
commit 8a862a2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/bamt/nodes/conditional_mixture_gaussian_node.py b/bamt/nodes/conditional_mixture_gaussian_node.py
@@ -163,12 +163,12 @@ def choose(
         pvals: parent values
         """
         mean, covariance, w = self.get_dist(node_info, pvals)
-
-        n_comp = len(w)
-
+
         if np.isnan(w):
             return np.nan
-
+
+        n_comp = len(w)
+
         gmm = GMM(
             n_components=n_comp,
             priors=w,