From dca7cb8e8ba99efc0fe8265acd9df42bcbd58f30 Mon Sep 17 00:00:00 2001 From: Mihir Kulkarni Date: Fri, 27 Dec 2024 20:59:36 +0100 Subject: [PATCH] removed dependence of gymutil.py from the training scripts Signed-off-by: Mihir Kulkarni --- aerial_gym/rl_training/cleanrl/ppo_continuous_action.py | 3 ++- aerial_gym/rl_training/rl_games/runner.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/aerial_gym/rl_training/cleanrl/ppo_continuous_action.py b/aerial_gym/rl_training/cleanrl/ppo_continuous_action.py index 8cde7ae..8d2408a 100644 --- 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 index fe09d7b..f343749 100644 --- 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