From 3774258cc7887f858025650662169c691a4784e2 Mon Sep 17 00:00:00 2001 From: Trinkle23897 <463003665@qq.com> Date: Thu, 11 Jun 2020 09:07:45 +0800 Subject: [PATCH] fix unittest --- test/base/test_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/base/test_env.py b/test/base/test_env.py index 5be5cbe..981e06f 100644 --- 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()