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