fix unittest

2020-06-11 09:07:45 +08:00 · 2020-06-11 09:07:45 +08:00 · 3774258cc7
commit 3774258cc7
parent 1a914336f7
1 changed files with 1 additions and 1 deletions
--- a/test/base/test_env.py
+++ b/test/base/test_env.py
@ -52,7 +52,7 @@ def test_vecenv(size=10, num=8, sleep=0.001):
    for v in venv:
        assert v.size == list(range(size, size + num))
        assert v.env_num == num
-        assert v.action_space == [Discrete(1)] * num
+        assert v.action_space == [Discrete(2)] * num

    for v in venv:
        v.close()