fix_memory_get_item

rmst · Nov 12, 2019 · 80429a0 · 80429a0
1 parent 7348fa6
commit 80429a0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/rtrl/memory.py b/rtrl/memory.py
@@ -37,6 +37,9 @@ def append(self, r, done, info, obs, action):
   def __len__(self):
     return len(self.memory)
 
+  def __getitem__(self, item):
+    return self.memory[item]
+
   def sample_indices(self):
     return (randint(0, len(self.memory) - 1) for _ in range(self.batchsize))