Merge branch 'main' of github.com:JamesKrW/vagen (ba29a44a) · Commits · 张泽凯 / VAGEN

README.md

+3 −0

Original line number	Diff line number	Diff line
		@@ -66,6 +66,9 @@ cd VAGEN
		bash scripts/install.sh
		```
		## Examples

		Note: VAGEN currently supports several environments: sokoban, frozenlake, svg, navigation, and primitive skill.
		For simplifying installation and execution, we have commented out all environments except sokoban and frozenlake. If you wish to run other environments, please uncomment the corresponding sections in `scripts/install.sh` and `vagen/env/__init__.py`.
		```
		# Login to wandb
		wandb login

scripts/install.sh

+9 −8

Original line number	Diff line number	Diff line
		@@ -32,13 +32,14 @@ echo "Installing Frozenlake dependencies"
		pip install 'gymnasium'
		pip install "gymnasium[toy-text]"

		echo "Installing Navigation dependencies"
		pip install ai2thor==5.0.0
		pip install numpy==1.25.1

		echo "Installing SVG dependencies"
		pip install "bs4"
		pip install "svgpathtools"
		pip install "cairosvg"
		# echo "Installing Navigation dependencies"
		# pip install ai2thor==5.0.0
		# pip install numpy==1.25.1

		# echo "Installing SVG dependencies"
		# pip install "bs4"
		# pip install "svgpathtools"
		# pip install "cairosvg"
		# pip install "dreamsim"

		echo "Installation complete!"
		No newline at end of file

vagen/env/README.md

+8 −0

Original line number	Diff line number	Diff line
		@@ -3,6 +3,12 @@
		# Start a General Server in debug mode
		python vagen/env/server.py --debug
		```

		### SVG
		```
		sudo apt-get update && sudo apt-get install -y libcairo2
		```

		### Navigation
		```

		@@ -19,6 +25,8 @@ sudo apt install vulkan-tools

		### Manipulation
		```
		#sudo apt-get update && sudo apt-get install -y libx11-6
		#sudo apt-get update && sudo apt-get install -y libgl1-mesa-glx
		pip install --upgrade mani_skill
		python -m mani_skill.utils.download_asset "PickSingleYCB-v1"
		python -m mani_skill.utils.download_asset partnet_mobility_cabinet

vagen/env/init.py

+20 −18

Original line number	Diff line number	Diff line
		from .sokoban import SokobanEnv,SokobanEnvConfig
		from .frozenlake import FrozenLakeEnv,FrozenLakeEnvConfig, FrozenLakeService
		from .navigation import NavigationEnv, NavigationEnvConfig, NavigationServiceConfig, NavigationService
		from .svg import SVGEnv, SvgEnvConfig, SVGService
		from .svg import SVGEnv, SvgEnvConfig, SVGService, SVGServiceConfig
		from .primitive_skill import PrimitiveSkillEnv, PrimitiveSkillEnvConfig, PrimitiveSkillService, PrimitiveSkillConfig

		REGISTERED_ENV = {
		"sokoban": {
		"env_cls": SokobanEnv,
		@@ -13,21 +14,22 @@ REGISTERED_ENV = {
		"config_cls": FrozenLakeEnvConfig,
		"service_cls": FrozenLakeService
		},
		"navigation": {
		"env_cls": NavigationEnv,
		"config_cls": NavigationEnvConfig,
		"service_cls": NavigationService,
		"service_config_cls": NavigationServiceConfig
		},
		"svg": {
		"env_cls": SVGEnv,
		"config_cls": SvgEnvConfig,
		"service_cls": SVGService
		},
		"primitive_skill": {
		"env_cls": PrimitiveSkillEnv,
		"config_cls": PrimitiveSkillEnvConfig,
		"service_cls": PrimitiveSkillService,
		"service_config_cls": PrimitiveSkillConfig
		}
		# "navigation": {
		# "env_cls": NavigationEnv,
		# "config_cls": NavigationEnvConfig,
		# "service_cls": NavigationService,
		# "service_config_cls": NavigationServiceConfig
		# },
		# "svg": {
		# "env_cls": SVGEnv,
		# "config_cls": SvgEnvConfig,
		# "service_cls": SVGService,
		# "service_config_cls": SVGServiceConfig
		# },
		# "primitive_skill": {
		# "env_cls": PrimitiveSkillEnv,
		# "config_cls": PrimitiveSkillEnvConfig,
		# "service_cls": PrimitiveSkillService,
		# "service_config_cls": PrimitiveSkillConfig
		# }
		}
		No newline at end of file

vagen/env/svg/init.py

+1 −0

Original line number	Diff line number	Diff line
		from .env_config import SvgEnvConfig
		from .env import SVGEnv
		from .service import SVGService
		from .service_config import SVGServiceConfig
		No newline at end of file

Admin message