Commit b1878e9f authored by Yamada Hiroyuki's avatar Yamada Hiroyuki
Browse files

Fix: Enlarge ubyt -> ushort to avoid overflow

parent d220ac6c

test/test_v8.py

+2 −2
Original line number Diff line number Diff line
@@ -117,7 +117,7 @@ class TestReplayBuffer(unittest.TestCase): 




        for i in range(512):

            obs = np.ones(obs_shape,dtype=np.ubyte) * i

            obs = np.ones(obs_shape, dtype=np.ushort) * i

            rb.add(obs=obs, act=act, rew=rew, next_obs=obs+1, done=done)



        sample = rb._encode_sample(range(buffer_size))