Hot fix for conditional_mixture_gaussian_node

Note that w can be either list or np.nan
aimclub · Oct 11, 2023 · f67ba4d · f67ba4d
1 parent 8a862a2
commit f67ba4d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/bamt/nodes/conditional_mixture_gaussian_node.py b/bamt/nodes/conditional_mixture_gaussian_node.py
@@ -163,8 +163,9 @@ def choose(
         pvals: parent values
         """
         mean, covariance, w = self.get_dist(node_info, pvals)
-
-        if np.isnan(w):
+
+        # check if w is nan or list of weights
+        if not isinstance(w, list):
             return np.nan
 
         n_comp = len(w)