12 lines
688 B
Python
12 lines
688 B
Python
from robogauge.utils.task_register import task_register
|
|
from robogauge.tasks.simulator.mujoco_config import MujocoConfig
|
|
from robogauge.tasks.robots import RobotConfig, Go2Config, Go2MoEConfig
|
|
from robogauge.tasks.pipeline import BasePipeline
|
|
from robogauge.tasks.gauge import BaseGaugeConfig
|
|
|
|
from robogauge.tasks.custom.go2_flat_task import Go2FlatGaugeConfig, Go2FlatConfig, Go2MoEFlatConfig
|
|
|
|
task_register.register('base', BasePipeline, MujocoConfig, BaseGaugeConfig, RobotConfig)
|
|
task_register.register('go2_flat', BasePipeline, MujocoConfig, Go2FlatGaugeConfig, Go2FlatConfig)
|
|
task_register.register('go2_moe_flat', BasePipeline, MujocoConfig, Go2FlatGaugeConfig, Go2MoEFlatConfig)
|