# -*- coding: utf-8 -*-
"""MoE policy configuration for the trained Go1 checkpoint."""

from robogauge.tasks.robots.go1.go1_config import Go1Config, Go1TerrainConfig


class Go1MoEConfig(Go1Config):
    robot_class = 'Go1MoE'


class Go1MoETerrainConfig(Go1TerrainConfig, Go1MoEConfig):
    robot_class = 'Go1MoE'