From 454be0d1ab0a0aaa929b45b5c1c1744fee145e02 Mon Sep 17 00:00:00 2001
From: Wong Yi Jie <55955482+yjwong1999@users.noreply.github.com>
Date: Wed, 14 Jun 2023 12:20:12 +0800
Subject: [PATCH] Update run_simulation.py

---
 run_simulation.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/run_simulation.py b/run_simulation.py
index 4a738be..5970ffb 100644
--- a/run_simulation.py
+++ b/run_simulation.py
@@ -14,14 +14,21 @@
 # validate the weight path
 if not os.path.isdir(STORE_PATH):
     raise NotImplementedError("The provided weight path does not exist!")
+
 # get DRL_ALGO
-if 'ddpg' in STORE_PATH:
+if 'td3' in STORE_PATH:
+    DRL_ALGO = 'td3'
+else:
     DRL_ALGO = 'ddpg'
+
+# reward design
+if 'see' in STORE_PATH:
+    REWARD_DESIGN = 'see'
 else:
-    DRL_ALGO = 'td3'
+    REWARD_DESIGN = 'ssr'
 
-REWARD_DESIGN = args.reward
-SEEDS = args.seeds
+# seeds and episode number
+SEEDS = None
 EPISODE_NUM = args.ep_num