fix config path

Author: mingxueg-nv

scripts/reinforcement_learning/rlinf/play_rlinf.py

@@ -57,8 +57,8 @@
 # Resolve config path and name from CLI args
 if not args_cli.config_name:
     parser.error("--config_name is required (e.g. --config_name isaaclab_ppo_gr00t_assemble_trocar)")
-config_dir = args_cli.config_path or str(SCRIPT_DIR)
 config_name = args_cli.config_name
+config_dir = cli_args.resolve_config_dir(config_name, args_cli.config_path)
 os.environ["RLINF_CONFIG_FILE"] = str(Path(config_dir) / f"{config_name}.yaml")
 
 # Add config dir to PYTHONPATH so that Ray rollout workers can resolve

scripts/reinforcement_learning/rlinf/train_rlinf.py

@@ -70,8 +70,8 @@ def run(argv: list[str]) -> None:
         _list_tasks()
         return
 
-    config_dir = args_cli.config_path or str(RLINF_DIR)
     config_name = args_cli.config_name
+    config_dir = CLI_ARGS.resolve_config_dir(config_name, args_cli.config_path)
     os.environ["RLINF_CONFIG_FILE"] = str(Path(config_dir) / f"{config_name}.yaml")
 
     if config_dir not in os.environ.get("PYTHONPATH", ""):