From f67ba4ddac3b1562b1b1e8c064746168cb51d701 Mon Sep 17 00:00:00 2001
From: Roman Netrogolov <68499591+Roman223@users.noreply.github.com>
Date: Wed, 11 Oct 2023 20:48:17 +0300
Subject: [PATCH] Hot fix for conditional_mixture_gaussian_node

Note that w can be either list or np.nan
---
 bamt/nodes/conditional_mixture_gaussian_node.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/bamt/nodes/conditional_mixture_gaussian_node.py b/bamt/nodes/conditional_mixture_gaussian_node.py
index ed2e32e..11d1df2 100644
--- 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)