Update HAC.py

nikhilbarhate99 · Nov 24, 2021 · 96bac54 · 96bac54
1 parent b3554a7
commit 96bac54
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/HAC.py b/HAC.py
@@ -104,15 +104,14 @@ def run_HAC(self, env, i_level, state, goal, is_subgoal_test):
                 next_state, rew, done, _ = env.step(action)
 
                 if self.render:
-                    # env.render()
+
+                    # env.render() ##########
 
                     if self.k_level == 2:
                         env.unwrapped.render_goal(self.goals[0], self.goals[1])
                     elif self.k_level == 3:
                         env.unwrapped.render_goal_2(self.goals[0], self.goals[1], self.goals[2])
 
-                    for _ in range(1000000):
-                        continue
 
                 # this is for logging
                 self.reward += rew