From c5cd186c06b1d7aa00dbc32b63ab63243493c0e0 Mon Sep 17 00:00:00 2001
From: Gaiejj <524339208@qq.com>
Date: Thu, 24 Aug 2023 20:40:04 +0800
Subject: [PATCH] docs: update README
---
README.md | 34 ++++++++++++++++++++++++++++++++++
safepo/evaluate.py | 2 +-
2 files changed, 35 insertions(+), 1 deletion(-)
diff --git a/README.md b/README.md
index df05c5c..3f47a00 100644
--- a/README.md
+++ b/README.md
@@ -108,6 +108,40 @@ For more details, please refer to [Safety-Gymnasium](https://github.com/PKU-Alig
| HalfCheetah, Hopper, Swimmer, Walker2d, Ant, Humanoid |
SafetyAntVelocity-v1 |
+
+ | Safe Multi-Agent |
+ MultiGoal[012] |
+ Multi-Point, Multi-Ant |
+ SafetyAntMultiGoal1-v0 |
+
+
+ | Multi-Agent Velocity |
+ 6x1HalfCheetah, 2x3HalfCheetah, 3x1Hopper, 2x1Swimmer, 2x3Walker2d, 2x4Ant, 4x2Ant, 9|8Humanoid |
+ Safety2x4AntVelocity-v0 |
+
+
+ | Safe Isaac Gym |
+ FreightFrankaCloseDrawer |
+ FreightFranka |
+ FreightFrankaCloseDrawer |
+
+
+ | FreightFrankaPickAndPlace |
+
+
+ | ShadowHandCatchOver2Underarm_Safe_finger |
+ ShadowHands |
+ ShadowHandCatchOver2Underarm_Safe_finger |
+
+
+ | ShadowHandCatchOver2Underarm_Safe_joint |
+
+
+ | ShadowHandOver_Safe_finger |
+
+
+ | ShadowHandOver_Safe_joint |
+
diff --git a/safepo/evaluate.py b/safepo/evaluate.py
index 58d84f1..dee85cb 100644
--- a/safepo/evaluate.py
+++ b/safepo/evaluate.py
@@ -137,7 +137,7 @@ def single_runs_eval(eval_dir, eval_episodes):
config_path = eval_dir + '/config.json'
config = json.load(open(config_path, 'r'))
env = config['task'] if 'task' in config.keys() else config['env_name']
- if env in multi_agent_velocity_map.keys() or multi_agent_goal_tasks:
+ if env in multi_agent_velocity_map.keys():
reward, cost = eval_multi_agent(eval_dir, eval_episodes)
else:
reward, cost = eval_single_agent(eval_dir, eval_episodes)