removed dependence of gymutil.py from the training scripts

Signed-off-by: Mihir Kulkarni <[email protected]>
ntnu-arl · Dec 27, 2024 · dca7cb8 · dca7cb8
1 parent 71bf152
commit dca7cb8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/aerial_gym/rl_training/cleanrl/ppo_continuous_action.py b/aerial_gym/rl_training/cleanrl/ppo_continuous_action.py
@@ -43,6 +43,7 @@
 
 
 from aerial_gym.registry.task_registry import task_registry
+from aerial_gym.utils.helpers import parse_arguments
 
 
 def get_args():
@@ -219,7 +220,7 @@ def get_args():
     ]
 
     # parse arguments
-    args = gymutil.parse_arguments(description="RL Policy", custom_parameters=custom_parameters)
+    args = parse_arguments(description="RL Policy", custom_parameters=custom_parameters)
 
     args.batch_size = int(args.num_envs * args.num_steps)
     args.minibatch_size = int(args.batch_size // args.num_minibatches)

diff --git a/aerial_gym/rl_training/rl_games/runner.py b/aerial_gym/rl_training/rl_games/runner.py
@@ -7,6 +7,7 @@
 
 
 from aerial_gym.registry.task_registry import task_registry
+from aerial_gym.utils.helpers import parse_arguments
 
 import gym
 from gym import spaces
@@ -249,7 +250,7 @@ def get_args():
     ]
 
     # parse arguments
-    args = gymutil.parse_arguments(description="RL Policy", custom_parameters=custom_parameters)
+    args = parse_arguments(description="RL Policy", custom_parameters=custom_parameters)
 
     # name allignment
     args.sim_device_id = args.compute_device_id