robits.agents package

Submodules

robits.agents.base_agent module

class robits.agents.base_agent.BaseAgent

Bases: ABC

agent_name: str

abstractmethod get_action(step, observation)

lang_goal: str

prepare_observation(obs, i, episode_length)

robits.agents.mello_agent module

class robits.agents.mello_agent.MelloAgent(device_addr, robot)

Bases: object

get_action() → ndarray

get_mello_data()

get_mello_joint_positions()

wait_for_pose()

class robits.agents.mello_agent.MelloConfig(max_delta: float = 0.25, initial_joint_positions: numpy.ndarray | None = None)

Bases: object

initial_joint_positions: ndarray | None = None

max_delta: float = 0.25

robits.agents.sam2act_agent module