Cond mixture bug (#85)

* serialization fix * fix * Update conditional_mixture_gaussian_node.py
aimclub · Oct 11, 2023 · bf1b476 · bf1b476
1 parent 1507472
commit bf1b476
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.isnan(w):
+            return np.nan
+
         gmm = GMM(
             n_components=n_comp,
             priors=w,