hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix unittest

Browse Source
This commit is contained in:
Trinkle23897 2020-06-11 09:07:45 +08:00
parent 1a914336f7
commit 3774258cc7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/base/test_env.py
View File

@ -52,7 +52,7 @@ def test_vecenv(size=10, num=8, sleep=0.001):
for v in venv: for v in venv:
assert v.size == list(range(size, size + num)) assert v.size == list(range(size, size + num))
assert v.env_num == num assert v.env_num == num
assert v.action_space == [Discrete(1)] * num assert v.action_space == [Discrete(2)] * num
for v in venv: for v in venv:
v.close() v.close()