comment complex envs (eb88a6d3) · Commits · 张泽凯 / VAGEN

scripts/install.sh

+9 −9

Original line number	Diff line number	Diff line
		@@ -32,14 +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"
		pip install "dreamsim"
		# 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/init.py

+19 −18

Original line number	Diff line number	Diff line
		@@ -3,6 +3,7 @@ from .frozenlake import FrozenLakeEnv,FrozenLakeEnvConfig, FrozenLakeService
		from .navigation import NavigationEnv, NavigationEnvConfig, NavigationServiceConfig, NavigationService
		from .svg import SVGEnv, SvgEnvConfig, SVGService, SVGServiceConfig
		from .primitive_skill import PrimitiveSkillEnv, PrimitiveSkillEnvConfig, PrimitiveSkillService, PrimitiveSkillConfig

		REGISTERED_ENV = {
		"sokoban": {
		"env_cls": SokobanEnv,
		@@ -13,22 +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,
		"service_config_cls": SVGServiceConfig
		},
		"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

Admin message