Handle both Environment Steps and Episodes on the subscriber side. (#99)

This commit is contained in:
Balaji Subramaniam authored and Scott Leishman committed 2018-11-15 14:42:21 -08:00
1 parent 3358e04a6a
commit 101c55d37d
4 files changed
+23 -12

No files matched your search

+2 -2
View File
@@ -659,9 +659,9 @@ class GraphManager(object):
self.handle_episode_ended()
self.reset_required = True
def fetch_from_worker(self, num_steps=0):
def fetch_from_worker(self, num_consecutive_playing_steps=None):
if hasattr(self, 'memory_backend'):
for transition in self.memory_backend.fetch(num_steps):
for transition in self.memory_backend.fetch(num_consecutive_playing_steps):
self.emulate_act_on_trainer(EnvironmentSteps(1), transition)
def setup_memory_backend(self) -> None: