You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
413 B
15 lines
413 B
from agent import Agent
|
|
from state import JointState
|
|
|
|
|
|
class Robot(Agent):
|
|
def __init__(self, config, section):
|
|
super(Robot, self).__init__(config, section)
|
|
|
|
def act(self, ob):
|
|
if self.policy is None:
|
|
raise AttributeError('Policy attribute has to be set!')
|
|
state = JointState(self.get_full_state(), ob)
|
|
action = self.policy.predict(state)
|
|
return action
|