sharing best rollout rather than just the state at the tree node

DecMCTS.py

@@ -162,8 +162,8 @@ class Tree:
         temp.pop(1, None)
 
         top_n_nodes = sorted(temp, key=temp.get, reverse=True)[:self.comm_n]
-        X = [self.graph.node[n]["state"] for n in top_n_nodes]
-        q = [self.graph.node[n]["mu"]**2 for n in top_n_nodes]
+        X = [self.graph.node[n]["best_rollout"] for n in top_n_nodes if self.graph.node[n]["N"]>0]
+        q = [self.graph.node[n]["mu"]**2 for n in top_n_nodes if self.graph.node[n]["N"]>0]
         self.my_act_dist = ActionDistribution(X,q)
         return True