fix

aimclub · Oct 11, 2023 · bfcabda · bfcabda
1 parent 4381c55
commit bfcabda
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bamt/nodes/conditional_mixture_gaussian_node.py b/bamt/nodes/conditional_mixture_gaussian_node.py
@@ -166,6 +166,9 @@ def choose(
 
         n_comp = len(w)
 
+        if np.isna(w):
+            return np.nan
+
         gmm = GMM(
             n_components=n_comp,
             priors=w,