gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

'CompositeAgent' object has no attribute 'handle_episode_ended' (#136)

Browse Source
This commit is contained in:
Ryan Peach
2018-12-05 04:28:16 -05:00
committed by Gal Leibovich
parent 436b16016e
commit 3c58ed740b
2 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
rl_coach/agents/agent_interface.py
View File

@@ -162,4 +162,14 @@ class AgentInterface(object):
(could be name of level manager or composite agent) (could be name of level manager or composite agent)
:return: collection of all agent savers :return: collection of all agent savers
""" """
raise NotImplementedError raise NotImplementedError("")
def handle_episode_ended(self) -> None:
"""
Make any changes needed when each episode is ended.
This includes incrementing counters, updating full episode dependent values, updating logs, etc.
This function is called right after each episode is ended.
:return: None
"""
raise NotImplementedError("")

7
rl_coach/agents/composite_agent.py
View File

@@ -305,9 +305,12 @@ class CompositeAgent(AgentInterface):
for agent in self.agents.values(): for agent in self.agents.values():
agent.phase = val agent.phase = val
def end_episode(self) -> None: def handle_episode_ended(self) -> None:
""" """
End an episode Make any changes needed when each episode is ended.
This includes incrementing counters, updating full episode dependent values, updating logs, etc.
This function is called right after each episode is ended.
:return: None :return: None
""" """
self.current_episode += 1 self.current_episode += 1