File tree Expand file tree Collapse file tree
scripts/reinforcement_learning/rsl_rl Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments