Skip to content

Commit 4b2b7ba

Browse files
committed
Fix unified RSL-RL preset selectors
1 parent d677461 commit 4b2b7ba

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

scripts/reinforcement_learning/rsl_rl/train_rsl_rl.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,8 @@ def _parse_args(argv: list[str]) -> argparse.Namespace:
6666
"""Parse RSL-RL training arguments."""
6767
from isaaclab.utils.string import list_intersection, string_to_callable
6868

69+
from isaaclab_tasks.utils import fold_preset_tokens, setup_preset_cli
70+
6971
parser = argparse.ArgumentParser(description="Train an RL agent with RSL-RL.")
7072
add_common_train_args(
7173
parser,
@@ -79,15 +81,16 @@ def _parse_args(argv: list[str]) -> argparse.Namespace:
7981
)
8082
CLI_ARGS.add_rsl_rl_args(parser)
8183
add_isaaclab_launcher_args(parser)
82-
args_cli, remaining_args = parser.parse_known_args(argv)
84+
args_cli, remaining_args = setup_preset_cli(parser, argv)
8385
enable_cameras_for_video(args_cli)
8486

8587
remaining_args_env_registration = None
8688
if args_cli.external_callback:
8789
external_callback_function = string_to_callable(args_cli.external_callback, separator=".")
8890
remaining_args_env_registration = external_callback_function()
8991

90-
set_hydra_args(list_intersection(remaining_args, remaining_args_env_registration))
92+
remaining_args = list_intersection(remaining_args, remaining_args_env_registration)
93+
set_hydra_args(fold_preset_tokens(remaining_args))
9194
return args_cli
9295

9396

0 commit comments

Comments
 (0)