n1 ready for test

Author: kew6688

challenge/onsite_competition/sdk/main.py

@@ -51,8 +51,8 @@ def main():
         obs = env.get_observation()
 
         # print("agent step...")
-        # action is a integer in [0, 3], agent return [{'action': int, 'ideal_flag': bool}]
-        action = agent.step(obs)[0]['action']  # only take the first env's action integer
+        # action is a integer in [0, 3], agent return [{'action': [int], 'ideal_flag': bool}] (same to internvla_n1 agent)
+        action = agent.step(obs)[0]['action'][0]  # only take the first env's action integer
 
         # print("env step...")
         env.step(action)

challenge/onsite_competition/sdk/test_agent.py

@@ -23,14 +23,15 @@ def test_agent(cfg_path=None, obs=None):
 
     agent = AgentClient(cfg.agent)
     for _ in range(10):
-        action = agent.step([obs])[0]['action']  # modify your agent to match the output format
+        action = agent.step([obs])[0]['action'][0]  # modify your agent to match the output format
         print(f"Action taken: {action}")
         assert action in [0, 1, 2, 3]
 
 
 if __name__ == "__main__":
     # use your own path
-    cfg_path = '/root/InternNav/scripts/eval/configs/h1_rdp_cfg.py'
+    # cfg_path = '/root/InternNav/scripts/eval/configs/h1_rdp_cfg.py'
+    cfg_path = '/root/InternNav/scripts/eval/configs/h1_internvla_n1_cfg.py'
     rs_meta_path = 'challenge/onsite_competition/captures/rs_meta.json'
 
     fake_obs_256 = {
@@ -51,5 +52,5 @@ def test_agent(cfg_path=None, obs=None):
 
     # work in progress, baseline model will be updated soon
     # test_agent(cfg_path=cfg_path, obs=fake_obs_256)
-    # test_agent(cfg_path='scripts/eval/configs/h1_internvla_n1_cfg.py', obs=fake_obs_640)
+    test_agent(cfg_path=cfg_path, obs=fake_obs_640)
     print("All tests passed!")