fixed replay suffix bug

IvLabs · Jun 16, 2021 · 3a548c4 · 3a548c4
1 parent f38be86
commit 3a548c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/atari/fixed_replay_buffer.py b/atari/fixed_replay_buffer.py
@@ -30,7 +30,7 @@ def __init__(self, data_dir, replay_suffix, *args, **kwargs):  # pylint: disable
     self.add_count = np.array(0)
     self._replay_suffix = replay_suffix
     if not self._loaded_buffers:
-      if replay_suffix:
+      if replay_suffix is not None:
         assert replay_suffix >= 0, 'Please pass a non-negative replay suffix'
         self.load_single_buffer(replay_suffix)
       else: