From 8749b02f447632210fe9f484d024e29145d96ea0 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 3 Dec 2020 07:15:39 +0100
Subject: [PATCH] fix
---
reinforcement_learning/ppo/ppo_agent.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/reinforcement_learning/ppo/ppo_agent.py b/reinforcement_learning/ppo/ppo_agent.py
index 21ce12d..54ba25f 100644
--- a/reinforcement_learning/ppo/ppo_agent.py
+++ b/reinforcement_learning/ppo/ppo_agent.py
@@ -35,7 +35,8 @@ class DataBuffers:
return self.memory.get(handle, [])
def push_transition(self, handle, transition):
- self.memory.update({handle: self.get_transitions(handle).append(transition)})
+ transitions = self.get_transitions(handle).append(transition)
+ self.memory.update({handle: transitions})
class PPOAgent(Policy):
--
GitLab