ficed leaderboard data paths

dfki-ric-underactuated-lab · Oct 9, 2023 · 48c651b · 48c651b
1 parent 49eba60
commit 48c651b
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 6 deletions.
diff --git a/leaderboard/simulation/compute_leaderboard.py b/leaderboard/simulation/compute_leaderboard.py
@@ -50,7 +50,12 @@
                 print(f"Simulating controller {f}")
                 compute_leaderboard_data(data_dir, f)
                 recompute_leaderboard = True
-            data_paths[mod.leaderboard_config["name"]] = mod.leaderboard_config
+
+            conf = mod.leaderboard_config
+            conf["csv_path"] = os.path.join(
+                data_dir, mod.leaderboard_config["csv_path"]
+            )
+            data_paths[mod.leaderboard_config["name"]] = conf
 
 
 save_to = f"{data_dir}/leaderboard.csv"

diff --git a/leaderboard/simulation/con_ddpg.py b/leaderboard/simulation/con_ddpg.py
@@ -10,7 +10,7 @@
 
 name = "ddpg"
 leaderboard_config = {
-    "csv_path": "data/" + name + "/sim_swingup.csv",
+    "csv_path": name + "/sim_swingup.csv",
     "name": name,
     "simple_name": "DDPG",
     "short_description": "RL Policy learned with Deep Deterministic Policy Gradient.",

diff --git a/leaderboard/simulation/con_dircol_tvlqr_lqr.py b/leaderboard/simulation/con_dircol_tvlqr_lqr.py
@@ -26,7 +26,7 @@
 
 name = "dircol_tvlqr_lqr"
 leaderboard_config = {
-    "csv_path": "data/" + name + "/sim_swingup.csv",
+    "csv_path": name + "/sim_swingup.csv",
     "name": name,
     "simple_name": "Direct collocation and TVLQR",
     "short_description": "Direct collocation trajectory stabilized with time-varying LQR.",

diff --git a/leaderboard/simulation/con_energyshaping_lqr.py b/leaderboard/simulation/con_energyshaping_lqr.py
@@ -23,7 +23,7 @@
 
 name = "energyshaping_lqr"
 leaderboard_config = {
-    "csv_path": "data/" + name + "/sim_swingup.csv",
+    "csv_path": name + "/sim_swingup.csv",
     "name": name,
     "simple_name": "Energy Shaping and LQR",
     "short_description": "Energy shaping for swingup and LQR for stabilization.",

diff --git a/leaderboard/simulation/con_ilqr_tvlqr_lqr.py b/leaderboard/simulation/con_ilqr_tvlqr_lqr.py
@@ -34,7 +34,7 @@
 
 name = "ilqr_tvlqr_lqr"
 leaderboard_config = {
-    "csv_path": "data/" + name + "/sim_swingup.csv",
+    "csv_path": name + "/sim_swingup.csv",
     "name": name,
     "simple_name": "iLQR and TVLQR",
     "short_description": "iLQR trajectory stabilized with time-varying LQR.",

diff --git a/leaderboard/simulation/con_sac.py b/leaderboard/simulation/con_sac.py
@@ -3,7 +3,7 @@
 
 name = "sac"
 leaderboard_config = {
-    "csv_path": "data/" + name + "/sim_swingup.csv",
+    "csv_path": name + "/sim_swingup.csv",
     "name": name,
     "simple_name": "SAC",
     "short_description": "RL Policy learned with Soft Actor Critic.",