Skip to content

Commit

Permalink
changes random actions scale in test_record_video.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Mayankm96 committed Apr 14, 2023
1 parent 5507c5b commit 5117a3c
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from omni.isaac.kit import SimulationApp

# launch the simulator
app_experience = f"{os.environ['EXP_PATH']}/omni.isaac.sim.python.gym.headless.kit"
app_experience = f"{os.environ['EXP_PATH']}/omni.isaac.sim.python.gym.headless.render.kit"
config = {"headless": True}
simulation_app = SimulationApp(config, experience=app_experience)

Expand Down Expand Up @@ -65,7 +65,7 @@ def test_record_video(self):
# parse configuration
env_cfg = parse_env_cfg(task_name, use_gpu=self.use_gpu, num_envs=self.num_envs)
# create environment
env = gym.make(task_name, cfg=env_cfg, headless=True, viewport=True)
env = gym.make(task_name, cfg=env_cfg, headless=self.headless, viewport=True)

# directory to save videos
videos_dir = os.path.join(self.videos_dir, task_name)
Expand All @@ -77,7 +77,7 @@ def test_record_video(self):
# reset environment
env.reset()
# simulate environment
for _ in range(600):
for _ in range(500):
# compute zero actions
actions = 2 * torch.rand((env.num_envs, env.action_space.shape[0]), device=env.device) - 1
# apply actions
Expand Down

0 comments on commit 5117a3c

Please sign in to comment.