Commit 7a6d0498 authored by wayne's avatar wayne
Browse files

fix: vlnce env service

parent de713088

vagen/env/vlnce/service.py

+3 −3
Original line number Diff line number Diff line
@@ -40,9 +40,9 @@ class VLNCEService(BaseService): 
        # Define worker function

        def create_single_env(env_id, config):

            # Verify environment type

            env_name = config.get('env_name', 'VLNCE')

            if env_name != 'VLNCE':

                return env_id, None, f"Expected environment type 'VLNCE', got '{env_name}'"

            env_name = config.get('env_name', 'vlnce')

            if env_name != 'vlnce':

                return env_id, None, f"Expected environment type 'vlnce', got '{env_name}'"

            

            env_config_dict = config['env_config']

            env_config = VLNCEEnvConfig(**env_config_dict)

vagen/env/vlnce/service_config.py

+1 −1
Original line number Diff line number Diff line
@@ -4,4 +4,4 @@ from typing import Dict, List, Tuple, Optional, Any, Union 


@dataclass

class VLNCEServiceConfig(BaseServiceConfig):

    pass
 
 No newline at end of file
 
    use_state_reward: bool = False

vagen/server/config/server.yaml

+3 −0
Original line number Diff line number Diff line
@@ -30,4 +30,7 @@ primitive_skill: 
sokoban:

  max_workers: 48

  use_state_reward: ${use_state_reward}

vlnce:

  max_workers: 48

  use_state_reward: False

CRA Git | Maintained and supported by SUSTech CRA and CCSE