| … | … | @@ -19,6 +19,7 @@ from rocca.agents import OpencogAgent |
| 19 | 19 | from rocca.agents.utils import agent_log |
| 20 | 20 | from rocca.envs.malmo_demo.chase_env import mission_xml |
| 21 | 21 | from rocca.envs.wrappers import MalmoWrapper |
| 22 | from opencog.ure import ure_logger |
| 22 | 23 | |
| 23 | 24 | ############### |
| 24 | 25 | # Chase Agent # |
| … | … | @@ -27,8 +28,8 @@ from rocca.envs.wrappers.utils import mk_action |
| 27 | 28 | |
| 28 | 29 | |
| 29 | 30 | class ChaseAgent(OpencogAgent): |
| 30 | | def __init__(self, env, action_space, p_goal, n_goal): |
| 31 | | OpencogAgent.__init__(self, env, action_space, p_goal, n_goal) |
| 31 | def __init__(self, env, atomspace, action_space, p_goal, n_goal): |
| 32 | OpencogAgent.__init__(self, env, atomspace, action_space, p_goal, n_goal) |
| 32 | 33 | env.step(mk_action("hotbar.9", 1)) # Press the hotbar key |
| 33 | 34 | env.step( |
| 34 | 35 | mk_action("hotbar.9", 0) |
| … | … | @@ -49,6 +50,14 @@ class ChaseAgent(OpencogAgent): |
| 49 | 50 | if __name__ == "__main__": |
| 50 | 51 | atomspace = AtomSpace() |
| 51 | 52 | set_default_atomspace(atomspace) |
| 53 | |
| 54 | # Init loggers |
| 55 | log.set_level("info") |
| 56 | # log.set_sync(True) |
| 57 | agent_log.set_level("debug") |
| 58 | # agent_log.set_sync(True) |
| 59 | ure_logger().set_level("debug") |
| 60 | |
| 52 | 61 | # Wrap environment |
| 53 | 62 | wrapped_env = MalmoWrapper(missionXML=mission_xml, validate=True) |
| 54 | 63 | |
| … | … | @@ -68,7 +77,7 @@ if __name__ == "__main__": |
| 68 | 77 | } |
| 69 | 78 | |
| 70 | 79 | # ChaseAgent |
| 71 | | ca = ChaseAgent(wrapped_env, action_space, pgoal, ngoal) |
| 80 | ca = ChaseAgent(wrapped_env, atomspace, action_space, pgoal, ngoal) |
| 72 | 81 | |
| 73 | 82 | # Eat some food. |
| 74 | 83 | ca.eat(4) |