update (9cbea459) · Commits · 张泽凯 / VAGEN

vagen/benchmark/alfworld/benchmark_config.yaml

+8 −9

Original line number	Diff line number	Diff line
		@@ -28,13 +28,12 @@ benchmark:
		output_dir: benchmark_results

		datasets:
		# ALFWorld with vision mode
		# - name: alfworld-vision
		# train_path: data/alfworld-vision-benchmark/train.parquet
		# test_path: data/alfworld-vision-benchmark/test.parquet
		# use_split: both

		- name: navigation
		train_path: data/navigation-benchmark/train.parquet
		test_path: data/navigation-benchmark/test.parquet
		- name: alfworld-vision
		train_path: data/alfworld-vision-benchmark/train.parquet
		test_path: data/alfworld-vision-benchmark/test.parquet
		use_split: both

		# - name: navigation
		# train_path: data/navigation-benchmark/train.parquet
		# test_path: data/navigation-benchmark/test.parquet
		# use_split: both
		No newline at end of file

vagen/benchmark/alfworld/dataset_config.yaml

+11 −12

Original line number	Diff line number	Diff line
		# ALFWorld with vision mode
		# alfworld-vision:
		# env_name: alfworld
		# env_config:
		# render_mode: vision
		# alf_config_path: "${env:ALFWORLD_DATA}/alf-config.yaml"
		# max_actions_per_step: 1
		# train_size: 8
		# test_size: 4

		navigation:
		env_name: navigation
		alfworld-vision:
		env_name: alfworld
		env_config:
		render_mode: vision
		alf_config_path: "${env:ALFWORLD_DATA}/alf-config.yaml"
		max_actions_per_step: 1
		train_size: 8
		test_size: 4

		# navigation:
		# env_name: navigation
		# env_config:
		# render_mode: vision
		# train_size: 8
		# test_size: 4
		No newline at end of file

vagen/env/init.py

+14 −14

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 .navigation import NavigationEnv, NavigationEnvConfig, NavigationServiceConfig, NavigationService
		# from .svg import SVGEnv, SvgEnvConfig, SVGService, SVGServiceConfig
		# from .primitive_skill import PrimitiveSkillEnv, PrimitiveSkillEnvConfig, PrimitiveSkillService, PrimitiveSkillServiceConfig
		# from .alfworld import ALFWorldEnv, ALFWorldEnvConfig, ALFWorldService, ALFWorldServiceConfig
		from .alfworld import ALFWorldEnv, ALFWorldEnvConfig, ALFWorldService, ALFWorldServiceConfig
		REGISTERED_ENV = {
		"sokoban": {
		"env_cls": SokobanEnv,
		@@ -14,12 +14,12 @@ REGISTERED_ENV = {
		"config_cls": FrozenLakeEnvConfig,
		"service_cls": FrozenLakeService
		},
		"navigation": {
		"env_cls": NavigationEnv,
		"config_cls": NavigationEnvConfig,
		"service_cls": NavigationService,
		"service_config_cls": NavigationServiceConfig
		},
		# "navigation": {
		# "env_cls": NavigationEnv,
		# "config_cls": NavigationEnvConfig,
		# "service_cls": NavigationService,
		# "service_config_cls": NavigationServiceConfig
		# },
		# "svg": {
		# "env_cls": SVGEnv,
		# "config_cls": SvgEnvConfig,
		@@ -32,10 +32,10 @@ REGISTERED_ENV = {
		# "service_cls": PrimitiveSkillService,
		# "service_config_cls": PrimitiveSkillServiceConfig
		# },
		# "alfworld": {
		# "env_cls": ALFWorldEnv,
		# "config_cls": ALFWorldEnvConfig,
		# "service_cls": ALFWorldService,
		# "service_config_cls": ALFWorldServiceConfig
		# },
		"alfworld": {
		"env_cls": ALFWorldEnv,
		"config_cls": ALFWorldEnvConfig,
		"service_cls": ALFWorldService,
		"service_config_cls": ALFWorldServiceConfig
		},
		}
		No newline at end of file

Admin message