fix(pu): fix unittest in test_ding_env_wrapper

opendilab · Dec 7, 2023 · 430e020 · 430e020
1 parent 45461a7
commit 430e020
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lzero/envs/tests/test_ding_env_wrapper.py b/lzero/envs/tests/test_ding_env_wrapper.py
@@ -26,6 +26,6 @@ def test(self):
 
         obs = ding_env.reset()
 
-        assert isinstance(obs, np.ndarray)
+        assert isinstance(obs[0], np.ndarray)
         action = ding_env.random_action()
         print('random_action: {}, action_space: {}'.format(action.shape, ding_env.action_space))