comment (faf90320) · Commits · 张泽凯 / VAGEN

vagen/env/init.py

+16 −16

Original line number	Diff line number	Diff line
		from .sokoban import SokobanEnv,SokobanConfig
		from .frozenlake import FrozenLakeEnv,FrozenLakeConfig, FrozenLakeService
		from .navigation import NavigationEnv, NavigationConfig
		from .svgdino import SVGDINOEnv, SVGDINOConfig
		from .svg import SVGEnv, SVGConfig, SVGService
		# from .navigation import NavigationEnv, NavigationConfig
		# from .svgdino import SVGDINOEnv, SVGDINOConfig
		# from .svg import SVGEnv, SVGConfig, SVGService

		REGISTERED_ENV = {
		"sokoban": {
		@@ -14,17 +14,17 @@ REGISTERED_ENV = {
		"config_cls": FrozenLakeConfig,
		"service_cls": FrozenLakeService
		},
		"navigation": {
		"env_cls": NavigationEnv,
		"config_cls": NavigationConfig
		},
		"svg": {
		"env_cls": SVGEnv,
		"config_cls": SVGConfig,
		"service_cls": SVGService
		},
		"svgdino": {
		"env_cls": SVGDINOEnv,
		"config_cls": SVGDINOConfig,
		}
		# "navigation": {
		# "env_cls": NavigationEnv,
		# "config_cls": NavigationConfig
		# },
		# "svg": {
		# "env_cls": SVGEnv,
		# "config_cls": SVGConfig,
		# "service_cls": SVGService
		# },
		# "svgdino": {
		# "env_cls": SVGDINOEnv,
		# "config_cls": SVGDINOConfig,
		# }
		}
		No newline at end of file