Add reward saving in an attribute #9

roboticslab-uc3m · Dec 18, 2023 · 6a55c47 · 6a55c47
1 parent 11dd93b
commit 6a55c47
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/line/gymnasium_playground_line/envs/line.py b/line/gymnasium_playground_line/envs/line.py
@@ -23,6 +23,9 @@ def __init__(self, render_mode=None, random_init=False, init_position=0., goal_p
         # current step
         self.current_step =0
 
+        # current reward
+        self.current_reward = 0
+
         # set max vel 
         self.max_w = 0.1 # 0.3rad/step or 18grad/step 
         self.max_dif = self.goal_position - self.initial_position
@@ -71,7 +74,8 @@ def step(self, action):
         penalization = 0.99**(self.current_step-1)
         reward *= penalization
 
-        self.writer.add_scalar('reward_each_step', reward)
+        self.current_reward =reward
+        # self.writer.add_scalar('reward_each_step', reward)
 
         terminated = False
         if dif < self.max_w:
@@ -87,3 +91,4 @@ def render(self):
         print("___________")
         print("Current position: ", self.current_position)
 
+