minor changes

montrealrobotics · Jul 31, 2023 · 5bf9bd2 · 5bf9bd2
1 parent cffbee1
commit 5bf9bd2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cleanrl/ppo_continuous_action.py b/cleanrl/ppo_continuous_action.py
@@ -87,7 +87,7 @@ def thunk():
             if rs_address:
                 env = gym.make(env_id, ur_model='ur5', rs_address=rs_address)
             elif target_ip:
-                env = gym.make(env_id, ur_model='ur5', ip=ip, gui=True)
+                env = gym.make(env_id, ur_model='ur5', ip=target_ip, gui=True)
             else:
                 env = gym.make(env_id)
         elif robot_type == "interbotix":
@@ -97,6 +97,8 @@ def thunk():
                 env = gym.make(env_id, robot_model='rx150', ip=target_ip, gui=True)
             else:
                 env = gym.make(env_id)
+        else:
+            env = gym.make(env_id)
 
         env = gym.wrappers.FlattenObservation(env)  # deal with dm_control's Dict observation space
         env = gym.wrappers.RecordEpisodeStatistics(env)
@@ -342,5 +344,4 @@ def get_action_and_value(self, x, action=None):
 
         print("SPS:", int(global_step / (time.time() - start_time)))
 
-
     envs.close()