robossembler/runtime
7
0
Fork 0
Code Issues 5 Pull requests Projects Releases Packages Wiki Activity Actions

rbs_gym multi camera and multi object setup

Browse source
This commit is contained in:
Ilya Uraev 2024-09-12 14:44:58 +03:00
parent b4b452297d
commit 1a562b05ca
34 changed files with 1082 additions and 816 deletions
Download patch file Download diff file Expand all files Collapse all files

9
env_manager/rbs_gym/scripts/test_agent.py
View file

@ -10,10 +10,13 @@ from rbs_gym import envs as gz_envs
from rbs_gym.utils.utils import StoreDict, str2bool
def main(args: Dict):
def main(args: argparse.Namespace):
# Create the environment
env = gym.make(args.env, **args.env_kwargs)
if args.env_kwargs is not None:
env = gym.make(args.env, **args.env_kwargs)
else:
env = gym.make(args.env)
# Initialize random seed
env.seed(args.seed)
@ -58,7 +61,7 @@ if __name__ == "__main__":
# Environment and its parameters
parser.add_argument(
"--env", type=str, default="Reach-Gazebo-v0", help="Environment ID"
"--env", type=str, default="Reach-ColorImage-Gazebo-v0", help="Environment ID"
)
parser.add_argument(
"--env-kwargs",