From 98692f2f9e5a3ff53f8f7c6edd1130b6421973f1 Mon Sep 17 00:00:00 2001
From: Lumi Pakkanen <lumi.pakkanen@gmail.com>
Date: Tue, 4 Sep 2018 22:21:05 +0300
Subject: [PATCH] Pass observation dtype to super class in
 WrappedPrioritizedReplayBuffer

---
 dopamine/replay_memory/prioritized_replay_buffer.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/dopamine/replay_memory/prioritized_replay_buffer.py b/dopamine/replay_memory/prioritized_replay_buffer.py
index cf1d0ab4..07430e1c 100644
--- a/dopamine/replay_memory/prioritized_replay_buffer.py
+++ b/dopamine/replay_memory/prioritized_replay_buffer.py
@@ -286,7 +286,8 @@ def __init__(self,
     memory = OutOfGraphPrioritizedReplayBuffer(
         observation_shape, stack_size, replay_capacity, batch_size,
         update_horizon, gamma, max_sample_attempts,
-        extra_storage_types=extra_storage_types)
+        extra_storage_types=extra_storage_types,
+        observation_dtype=observation_dtype)
     super(WrappedPrioritizedReplayBuffer, self).__init__(
         observation_shape,
         stack_size,
@@ -296,7 +297,8 @@ def __init__(self,
         update_horizon,
         gamma,
         wrapped_memory=memory,
-        extra_storage_types=extra_storage_types)
+        extra_storage_types=extra_storage_types,
+        observation_dtype=observation_dtype)
 
   def tf_set_priority(self, indices, priorities):
     """Sets the priorities for the given indices.