changed default env setting

Author: JamesKrW

vagen/env/frozenlake/env_config.py

@@ -12,10 +12,7 @@ class FrozenLakeEnvConfig(BaseEnvConfig):
     render_mode: str = "vision"  # "text" or "vision"
     max_actions_per_step: int = 3
     min_actions_to_succeed: int = 5
-    prompt_format: str = "free_think" 
-    # "free_think", "no_think", "grounding", "worldmodeling", "grounding_worldmodeling"
-    # "grounding_symbolic", "worldmodeling_symbolic", "grounding_worldmodeling_symbolic"
-    # "grounding_structured", "worldmodeling_structured", "grounding_worldmodeling_structured"
+    prompt_format: str = "grounding_worldmodeling"
 
     # configs for process reward for grounding and world modeling
     use_state_reward: bool = False

vagen/env/navigation/env_config.py

@@ -15,7 +15,7 @@ class NavigationEnvConfig(BaseEnvConfig):
     max_action_penalty: float = -0.1
     format_reward: float = 0.5
     gpu_device: int = 0
-    prompt_format: str = "free_think" 
+    prompt_format: str = "grounding_worldmodeling" 
     success_threshold: float = 1.5
     step_length: float = 0.5
     # "free_think", "no_think", "grounding", "worldmodeling", "grounding_worldmodeling"

vagen/env/primitive_skill/env_config.py

@@ -12,7 +12,7 @@ class PrimitiveSkillEnvConfig(BaseEnvConfig):
     record_video: bool = field(default=False)
     video_record_dir: str = field(default='./test')
     mask_success: bool = field(default=True)
-    prompt_format: str = "free_think" 
+    prompt_format: str = "grounding_worldmodeling" 
     # "free_think", "no_think", "grounding", "worldmodeling", "grounding_worldmodeling",
     # "grounding_structured","worldmodeling_structured","grounding_worldmodeling_structured"
 

vagen/env/sokoban/env_config.py

@@ -10,7 +10,7 @@ class SokobanEnvConfig(BaseEnvConfig):
     render_mode: str = "vision" # "vision" or "text"
     min_actions_to_succeed: int = 5
     max_actions_per_step: int = 3
-    prompt_format: str = "free_think" 
+    prompt_format: str = "grounding_worldmodeling" 
     # "free_think", "no_think", "grounding", "worldmodeling", "grounding_worldmodeling"
     # "grounding_symbolic", "worldmodeling_symbolic", "grounding_worldmodeling_symbolic"
     # "grounding_structured", "worldmodeling_structured", "grounding_worldmodeling_structured"

vagen/env/svg/env_config.py

@@ -22,7 +22,7 @@ class SvgEnvConfig(BaseEnvConfig):
     format_reward: float = 0.5
     format_penalty: float = 0.0
     # Prompt configuration
-    prompt_format: str = "free_think" 
+    prompt_format: str = "grounding_worldmodeling" 
     # "free_think", "no_think", "grounding", "worldmodeling", "grounding_worldmodeling"
     use_state_reward: bool = False