Commit 7cbb61f4 authored by jameskrw's avatar jameskrw
Browse files

minor

parent ba29a44a

vagen/env/primitive_skill/__init__.py

+1 −1
Original line number Diff line number Diff line 
from .env import PrimitiveSkillEnv

from .env_config import PrimitiveSkillEnvConfig

from .service import PrimitiveSkillService

from .service_config import PrimitiveSkillConfig
 
 No newline at end of file
 
from .service_config import PrimitiveSkillServiceConfig
 
 No newline at end of file

vagen/env/primitive_skill/service_config.py

+1 −1
Original line number Diff line number Diff line
@@ -2,7 +2,7 @@ from vagen.env.base.base_service_config import BaseServiceConfig 
from dataclasses import dataclass, fields,field



@dataclass

class PrimitiveSkillConfig(BaseServiceConfig):

class PrimitiveSkillServiceConfig(BaseServiceConfig):

    max_process_workers: int = field(default=8)

    max_thread_workers: int = field(default=4)

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

CRA Git | Maintained and supported by SUSTech CRA and CCSE