From fba389283ff8f79c1448f28bbd35fc23b0acebf0 Mon Sep 17 00:00:00 2001
From: wjxgeorge <12130815d@connect.polyu.hk>
Date: Sat, 14 Dec 2024 01:20:10 -0500
Subject: [PATCH] update test_envs.py

---
 tests/test_envs.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/tests/test_envs.py b/tests/test_envs.py
index ccc0bf32..64466c03 100644
--- a/tests/test_envs.py
+++ b/tests/test_envs.py
@@ -189,12 +189,8 @@ def test_robot_env_reset(spec):
     def _test_initial_states(env, seed=None):
         diag_dict = {}
 
-        init_obs = env.reset(seed=seed)
+        env.reset(seed=seed)
 
-        if isinstance(init_obs[0], dict):
-            diag_dict.update(**init_obs[0])
-        elif isinstance(init_obs[0], np.ndarray):
-            diag_dict.update({"observation": init_obs[0]})
         diag_dict.update(
             {
                 "qpos": env.unwrapped.data.qpos,
@@ -227,4 +223,3 @@ def _test_initial_states(env, seed=None):
 
     _test_initial_states(cur_env, seed=24)
     _test_initial_states(cur_env, seed=10)
-    return