Functions — github.com/mujocolab/mjlab

CodeHub MCPcopy Create free account

hub / github.com/mujocolab/mjlab / functions

Functions2,812 in github.com/mujocolab/mjlab

⨍ Functions 2,812◇ Types & classes 316↳ Endpoints 15

Method__contains__

Return True if a term named ``name`` is registered.

src/mjlab/managers/recorder_manager.py:165

Method__contains__

Always returns False since there are no terms.

src/mjlab/managers/recorder_manager.py:244

(self, other: Any)

src/mjlab/sim/sim_data.py:148

(self, other: Any)

src/mjlab/sim/sim_data.py:163

Method__getattr__

Delegate attribute access to wrapped environment.

src/mjlab/utils/wrappers/video_recorder.py:72

Method__getattr__

Delegate attribute access to the underlying tensor.

src/mjlab/sim/sim_data.py:75

Method__getattr__

Get attribute from the wrapped data, wrapping Warp arrays as TorchArray.

src/mjlab/sim/sim_data.py:201

Method__getattr__

(self, name: str)

src/mjlab/viewer/native/viewer.py:112

Method__getitem__

(self, key: str)

src/mjlab/scene/scene.py:145

Method__getitem__

Retrieve lagged frames per batch (LIFO). Args: key: Per-batch lags (Tensor) or shared lag (int). Shape (batch_size,) or scalar.

src/mjlab/utils/buffers/circular_buffer.py:219

Method__getitem__

Get item(s) from the tensor using standard indexing.

src/mjlab/sim/sim_data.py:63

Method__getitem__

(self, idx: int)

src/mjlab/viewer/native/viewer.py:108

(self, other: Any)

src/mjlab/sim/sim_data.py:160

(self, scene_cfg: SceneCfg, device: str)

src/mjlab/scene/scene.py:53

(self, cfg: NanGuardCfg, num_envs: int, mj_model: mujoco.MjModel)

src/mjlab/utils/nan_guard.py:34

( self, env: ManagerBasedRlEnv, video_folder: str | Path, episode_trigger: Callable[[int], boo

src/mjlab/utils/wrappers/video_recorder.py:43

( self, noise_model_cfg: noise_cfg.NoiseModelWithAdditiveBiasCfg, num_envs: int, device: str,

src/mjlab/utils/noise/noise_model.py:39

( self, min_lag: int = 0, max_lag: int = 3, batch_size: int = 1, device: str = "cpu",

src/mjlab/utils/buffers/delay_buffer.py:120

(self, max_len: int, batch_size: int, device: str)

src/mjlab/utils/buffers/circular_buffer.py:124

( self, num_envs: int, cfg: SimulationCfg, model: mujoco.MjModel | None = None, device: st

src/mjlab/sim/sim.py:194

Initialize the tensor proxy with a Warp array.

src/mjlab/sim/sim_data.py:18

(self, struct: T, nworld: int | None = None)

src/mjlab/sim/sim_data.py:196

( self, model: mujoco.MjModel, cfg: ViewerConfig, scene: Scene, sim_model: Any | None = No

src/mjlab/viewer/offscreen_renderer.py:19

( self, env: EnvProtocol, policy: PolicyProtocol, frame_rate: float = 30.0, verbosity: int

src/mjlab/viewer/base.py:153

(self, env_idx: int = 0, meansize: float = 0.1)

src/mjlab/viewer/debug_visualizer.py:211

( self, env: EnvProtocol, policy: PolicyProtocol, frame_rate: float = 60.0, key_callback:

src/mjlab/viewer/native/viewer.py:122

Initialize the MuJoCo native visualizer. Args: scn: MuJoCo scene to add visualizations to mj_model: MuJoCo model for creating visuali

src/mjlab/viewer/native/visualizer.py:19

( self, server: viser.ViserServer, camera_sensor: CameraSensor, mj_model: mujoco.MjModel,

src/mjlab/viewer/viser/camera_viewer.py:36

( self, env: EnvProtocol, policy: PolicyProtocol, frame_rate: float = 60.0, verbosity: Ver

src/mjlab/viewer/viser/viewer.py:67

Initialize the plotter. Args: server: The Viser server instance term_names: List of term names to plot name: Name prefix for th

src/mjlab/viewer/viser/term_plotter.py:46

( self, server: viser.ViserServer, mj_model: mujoco.MjModel, num_envs: int, sim_model: Any

src/mjlab/viewer/viser/scene.py:225

Initialize the reward bar panel. Args: server: The Viser server instance. term_names: List of reward term names. update_dt: Tim

src/mjlab/viewer/viser/reward_bar_panel.py:19

( self, input_dim: tuple[int, int], input_channels: int, temperature: float = 1.0, **cnn_k

src/mjlab/rl/spatial_softmax.py:80

( self, obs: TensorDict, obs_groups: dict[str, list[str]], obs_set: str, output_dim: int,

src/mjlab/rl/spatial_softmax.py:129

( self, env: VecEnv, train_cfg: dict, log_dir: str | None = None, device: str = "cpu", )

src/mjlab/rl/runner.py:16

( self, env: ManagerBasedRlEnv, clip_actions: float | None = None, )

src/mjlab/rl/vecenv_wrapper.py:10

(self, cfg: dict[str, TerminationTermCfg], env: ManagerBasedRlEnv)

src/mjlab/managers/termination_manager.py:36

(self, cfg: dict[str, CurriculumTermCfg], env: ManagerBasedRlEnv)

src/mjlab/managers/curriculum_manager.py:39

src/mjlab/managers/curriculum_manager.py:135

(self, env: ManagerBasedRlEnv)

src/mjlab/managers/manager_base.py:65

(self, env: ManagerBasedRlEnv)

src/mjlab/managers/manager_base.py:97

(self, cfg: dict[str, RecorderTermCfg], env: ManagerBasedRlEnv)

src/mjlab/managers/recorder_manager.py:151

src/mjlab/managers/recorder_manager.py:234

(self, cfg: dict[str, EventTermCfg], env: ManagerBasedRlEnv)

src/mjlab/managers/event_manager.py:153

(self, cfg: dict[str, ActionTermCfg], env: ManagerBasedRlEnv)

src/mjlab/managers/action_manager.py:79

( self, cfg: dict[str, RewardTermCfg], env: ManagerBasedRlEnv, *, scale_by_dt: bool = True

src/mjlab/managers/reward_manager.py:53

(self, cfg: dict[str, CommandTermCfg], env: ManagerBasedRlEnv)

src/mjlab/managers/command_manager.py:151

src/mjlab/managers/command_manager.py:279

(self, cfg: dict[str, MetricsTermCfg], env: ManagerBasedRlEnv)

src/mjlab/managers/metrics_manager.py:50

src/mjlab/managers/metrics_manager.py:203

(self, cfg: dict[str, ObservationGroupCfg], env)

src/mjlab/managers/observation_manager.py:123

( self, cfg: LearnedMlpActuatorCfg, entity: Entity, target_ids: list[int], target_names: l

src/mjlab/actuator/learned_actuator.py:82

( self, cfg: XmlActuatorCfg, entity: Entity, target_ids: list[int], target_names: list[str

src/mjlab/actuator/xml_actuator.py:49

( self, cfg: BuiltinPositionActuatorCfg, entity: Entity, target_ids: list[int], target_nam

src/mjlab/actuator/builtin_actuator.py:68

( self, cfg: BuiltinMotorActuatorCfg, entity: Entity, target_ids: list[int], target_names:

src/mjlab/actuator/builtin_actuator.py:216

( self, cfg: BuiltinVelocityActuatorCfg, entity: Entity, target_ids: list[int], target_nam

src/mjlab/actuator/builtin_actuator.py:517

( self, cfg: BuiltinMuscleActuatorCfg, entity: Entity, target_ids: list[int], target_names

src/mjlab/actuator/builtin_actuator.py:591

( self, cfg: DcMotorCfgT, entity: Entity, target_ids: list[int], target_names: list[str],

src/mjlab/actuator/dc_actuator.py:90

( self, cfg: ActuatorCfgT, entity: Entity, target_ids: list[int], target_names: list[str],

src/mjlab/actuator/actuator.py:154

( self, cfg: IdealPdCfgT, entity: Entity, target_ids: list[int], target_names: list[str],

src/mjlab/actuator/pd_actuator.py:41

( self, cfg: ManagerBasedRlEnvCfg, device: str, render_mode: str | None = None, **kwargs,

src/mjlab/envs/manager_based_rl_env.py:177

(self, cfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/events.py:434

(self, cfg: RewardTermCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/rewards.py:106

(self, cfg: RewardTermCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/rewards.py:136

(self, cfg: CurriculumTermCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/curriculums.py:133

(self, cfg: CurriculumTermCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/curriculums.py:172

(self, cfg: JointPositionActionCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/actions/actions.py:215

(self, cfg: JointVelocityActionCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/actions/actions.py:263

(self, cfg: JointEffortActionCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/actions/actions.py:278

(self, cfg: TendonLengthActionCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/actions/actions.py:328

(self, cfg: TendonVelocityActionCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/actions/actions.py:340

(self, cfg: TendonEffortActionCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/actions/actions.py:352

(self, cfg: SiteEffortActionCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/actions/actions.py:380

(self, cfg: DifferentialIKActionCfg, env: ManagerBasedRlEnv)

src/mjlab/envs/mdp/actions/differential_ik.py:116

( self, motion_file: str, input_fps: int, output_fps: int, device: torch.device | str,

src/mjlab/scripts/csv_to_npz.py:24

(self, dump_path: str | Path)

src/mjlab/scripts/nan_viz.py:26

src/mjlab/sensor/sensor.py:53

( self, mj_model: mujoco.MjModel, model: mjwarp.Model, data: mjwarp.Data, camera_sensors:

src/mjlab/sensor/sensor_context.py:47

(self, cfg: ContactSensorCfg)

src/mjlab/sensor/contact_sensor.py:229

(self, cfg: CameraSensorCfg)

src/mjlab/sensor/camera_sensor.py:139

( self, cfg: BuiltinSensorCfg | None = None, name: str | None = None )

src/mjlab/sensor/builtin_sensor.py:279

(self, cfg: RayCastSensorCfg)

src/mjlab/sensor/raycast_sensor.py:431

(self, cfg: TerrainEntityCfg, device: str)

src/mjlab/terrains/terrain_entity.py:129

(self, cfg: TerrainGeneratorCfg, device: str = "cpu")

src/mjlab/terrains/terrain_generator.py:149

(self, cfg: RewardTermCfg, env: ManagerBasedRlEnv)

src/mjlab/tasks/velocity/mdp/rewards.py:76

(self, cfg: RewardTermCfg, env: ManagerBasedRlEnv)

src/mjlab/tasks/velocity/mdp/rewards.py:274

(self, cfg: RewardTermCfg, env: ManagerBasedRlEnv)

src/mjlab/tasks/velocity/mdp/rewards.py:401

(self, cfg: UniformVelocityCommandCfg, env: ManagerBasedRlEnv)

src/mjlab/tasks/velocity/mdp/velocity_command.py:28

( self, cfg: MultiCubeLiftingCommandCfg, env: ManagerBasedRlEnv, )

src/mjlab/tasks/manipulation/mdp/commands.py:123

(self, cfg: MotionCommandCfg, env: ManagerBasedRlEnv)

src/mjlab/tasks/tracking/mdp/commands.py:67

( self, env: VecEnv, train_cfg: dict, log_dir: str | None = None, device: str = "cpu",

src/mjlab/tasks/tracking/rl/runner.py:50

(self, cfg: EntityCfg)

src/mjlab/entity/entity.py:146

scripts/demos/raycast_sensor.py:209

(self, scene, sim, device)

tests/test_projected_gravity_sensor.py:61

(self, unwrapped: Any)

tests/test_viser_update_policy.py:32

tests/test_foot_height_sensor.py:71

(self, cfg: RewardTermCfg, env)

tests/test_rewards.py:49

(self, cfg: RewardTermCfg, env)

tests/test_rewards.py:66

(self, **kwargs)

tests/test_variants.py:1286

tests/test_variants.py:1300

← previous next →901–1,000 of 2,812, ranked by callers