提交 ff31a86b 编写于 作者: N niuyazhe

fix(nyz): fix PER indice repeat unittest bug

上级 f089d02a
......@@ -88,7 +88,7 @@ def test_priority():
buffer = DequeBuffer(size=10)
buffer.use(PriorityExperienceReplay(buffer, buffer_size=10, IS_weight=True))
for _ in range(N):
buffer.push(get_data())
buffer.push(get_data(), meta={'priority': 2.0})
assert buffer.count() == N
for _ in range(N):
buffer.push(get_data(), meta={'priority': 2.0})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册