Skip to content
Snippets Groups Projects
Commit a911027e authored by Jayant Khatkar's avatar Jayant Khatkar
Browse files

quick fix

parent 207ba654
Branches
Tags
No related merge requests found
......@@ -16,6 +16,7 @@ from DecMCTS import Tree
# data can be anything required to calculate your
# global reward and available actions
# It can be in any format (only used by your reward and avail_actions functions)
data = {}
# Create an available actions function
......@@ -31,7 +32,7 @@ def avail_actions(data, state):
# actions taken by the current robot, and every other robot
# State is a dictionary with keys being robot IDs, and values
# are a list of actions taken from the starting position
def reward(dat, state):
def reward(data, state):
each_robot_sum = [sum(state[a]) for a in state]
return sum(each_robot_sum)
......
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