Commit cbae0a49 authored by Jayant Khatkar's avatar Jayant Khatkar

cleanup

parent 2792282d
......@@ -190,8 +190,11 @@ class Tree:
Init step.
"""
robot_state = self.graph.node[start_node]["state"]
options = self.available_actions(self.data, robot_state, self.id)
options = self.available_actions(
self.data,
self.graph.node[start_node]["state"],
self.id
)
# create empty nodes underneath the node being expanded
for o in options:
......@@ -206,8 +209,6 @@ class Tree:
return True
def grow(self, nsims=10, gamma = 0.9):
"""
Grow Tree by one node
......@@ -254,6 +255,7 @@ class Tree:
def send_comms(self):
return self.my_act_dist
def receive_comms(self, comms_in, robot_id):
"""
Save data which has been communicated to this tree
......@@ -268,3 +270,4 @@ class Tree:
"""
self.comms[robot_id] = comms_in
return True
......@@ -2,6 +2,6 @@ from distutils.core import setup
setup(
name='DecMCTS',
version='0.2dev',
version='0.3dev',
py_modules=['DecMCTS']
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment