Commit f14cd04a authored by jameskrw's avatar jameskrw
Browse files

updated service config

parent 18901cf6

vagen/env/__init__.py

+3 −2
Original line number Diff line number Diff line 
# First, import the modules that are assumed to be always available

from .sokoban import SokobanEnv, SokobanEnvConfig, SokobanService, SokobanServiceConfig

from .frozenlake import FrozenLakeEnv, FrozenLakeEnvConfig, FrozenLakeService

from .frozenlake import FrozenLakeEnv, FrozenLakeEnvConfig, FrozenLakeService, FrozenLakeServiceConfig



REGISTERED_ENV = {

    "sokoban": {
@@ -12,7 +12,8 @@ REGISTERED_ENV = { 
    "frozenlake": {

        "env_cls": FrozenLakeEnv,

        "config_cls": FrozenLakeEnvConfig,

        "service_cls": FrozenLakeService

        "service_cls": FrozenLakeService,

        "service_config_cls": FrozenLakeServiceConfig

    }

}

vagen/env/frozenlake/__init__.py

+2 −1
Original line number Diff line number Diff line 
from .env_config import FrozenLakeEnvConfig

from .env import FrozenLakeEnv

from .service import FrozenLakeService

from .service_config import FrozenLakeServiceConfig
 
 No newline at end of file

vagen/env/frozenlake/service_config.py

+2 −1
Original line number Diff line number Diff line
@@ -6,3 +6,4 @@ from typing import Dict, List, Tuple, Optional, Any, Union 
class FrozenLakeServiceConfig(BaseServiceConfig):

    preload_reward_model: Dict[str, Any] = field(default_factory=lambda: {"clip": False})

    device: Dict[str, Any] = field(default_factory=lambda: {"clip": 0})

    use_state_reward: bool = False
 
 No newline at end of file

vagen/env/navigation/service_config.py

+2 −1
Original line number Diff line number Diff line
@@ -4,3 +4,4 @@ from dataclasses import dataclass, fields,field 
@dataclass

class NavigationServiceConfig(BaseServiceConfig):

    devices: list = field(default_factory=lambda: [0])

    use_state_reward: bool = False
 
 No newline at end of file

vagen/env/primitive_skill/service_config.py

+2 −1
Original line number Diff line number Diff line
@@ -8,3 +8,4 @@ class PrimitiveSkillServiceConfig(BaseServiceConfig): 
    devices: list = field(default_factory=lambda: [0,1])

    spawn_method: str = field(default="fork")

    timeout: int = field(default=120)

    use_state_reward: bool = False
 
 No newline at end of file

CRA Git | Maintained and supported by SUSTech CRA and CCSE