diff --git a/embodichain/lab/gym/utils/gym_utils.py b/embodichain/lab/gym/utils/gym_utils.py index 20ac316a..b4a2054b 100644 --- a/embodichain/lab/gym/utils/gym_utils.py +++ b/embodichain/lab/gym/utils/gym_utils.py @@ -769,6 +769,12 @@ def add_env_launcher_args_to_parser(parser: argparse.ArgumentParser) -> None: default=False, action="store_true", ) + parser.add_argument( + "--arena_space", + help="The size of the arena space.", + default=5.0, + type=float, + ) parser.add_argument( "--enable_rt", help="Whether to use RTX rendering backend for the simulation.", @@ -829,6 +835,7 @@ def merge_args_with_gym_config(args: argparse.Namespace, gym_config: dict) -> di merged_config["headless"] = args.headless merged_config["enable_rt"] = args.enable_rt merged_config["gpu_id"] = args.gpu_id + merged_config["arena_space"] = args.arena_space return merged_config @@ -871,6 +878,7 @@ def build_env_cfg_from_args( sim_device=gym_config["device"], enable_rt=gym_config["enable_rt"], gpu_id=gym_config["gpu_id"], + arena_space=gym_config["arena_space"], ) return cfg, gym_config, action_config