From 29b62779e4991fc8ed1d52ecc95ca3fef46bb0af Mon Sep 17 00:00:00 2001 From: Jordi Salvador Date: Fri, 1 Apr 2022 15:56:06 +0200 Subject: [PATCH] Fix for memoryless agents --- allenact/algorithms/onpolicy_sync/storage.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/allenact/algorithms/onpolicy_sync/storage.py b/allenact/algorithms/onpolicy_sync/storage.py index 1519b59e7..504b29a9f 100644 --- a/allenact/algorithms/onpolicy_sync/storage.py +++ b/allenact/algorithms/onpolicy_sync/storage.py @@ -178,7 +178,7 @@ def initialize( **kwargs, ): if self.memory_specification is None: - self.memory_specification = recurrent_memory_specification + self.memory_specification = recurrent_memory_specification or {} self.action_space = action_space self.memory_first_last: Memory = self.create_memory(