Fixing a typo

little-nem · Jun 4, 2019 · 2fc60b9 · 2fc60b9
1 parent b287fcb
commit 2fc60b9
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/main.py b/main.py
@@ -76,8 +76,6 @@
 
         episode.add_step(action[0], new_obs, step_reward, achieved, terminal = done)
 
-        total_reward += step_reward
-
         current_obs_dict = new_obs_dict
 
     # store the episode in the replay buffer
@@ -197,8 +195,6 @@
 
                 episode.add_step(action[0], new_obs, step_reward, achieved, terminal=done)
 
-                total_reward += step_reward
-
                 current_obs_dict = new_obs_dict
 
             if info['is_success'] > 0.0: