MCPcopy Create free account

hub / github.com/mujocolab/mjlab / functions

Functions2,812 in github.com/mujocolab/mjlab

Methodnum_pairs
(self)
src/mjlab/entity/entity.py:497
Methodnum_rays
Total number of rays (num_frames * num_rays_per_frame).
src/mjlab/sensor/raycast_sensor.py:590
Methodnum_rays_per_frame
Number of rays per attachment frame.
src/mjlab/sensor/raycast_sensor.py:600
Methodnum_sites
(self)
src/mjlab/entity/entity.py:477
Methodnum_sites
(self)
tests/test_scene_entity_config.py:32
Methodnum_targets
Number of targets. ``ctrl_ids`` is laid out as ``[pos..., vel...]``, each block of length ``num_targets``.
src/mjlab/actuator/builtin_actuator.py:148
Methodnum_tendons
(self)
src/mjlab/entity/entity.py:481
Functionobject_to_goal_distance
Distance vector from object to goal in base frame.
src/mjlab/tasks/manipulation/mdp/observations.py:39
Functionobs_A
(env)
tests/test_observation_history.py:557
Functionobs_B
(env)
tests/test_observation_history.py:561
Functionobs_func
(env)
tests/test_observation_noise.py:39
Functionobs_func
(env)
tests/test_observation_history.py:37
Functionobs_func
(env)
tests/test_observation_delay.py:37
Functionobs_func2
(env)
tests/test_observation_history.py:459
Functionobs_func2
(env)
tests/test_observation_delay.py:483
Functionobs_func_a
(env)
tests/test_observation_noise.py:282
Functionobs_func_b
(env)
tests/test_observation_noise.py:285
Functionobs_with_nan
(env)
tests/test_observation_nan_handling.py:35
Functionobs_with_neginf
(env)
tests/test_observation_nan_handling.py:410
Methodobservation_space
(self)
src/mjlab/rl/vecenv_wrapper.py:36
Methodoffset
Action offset.
src/mjlab/envs/mdp/actions/actions.py:130
Methodon_viewer_pause
Notify all command terms of viewer pause state change.
src/mjlab/managers/command_manager.py:191
Methodon_viewer_pause
(self, paused: bool)
src/mjlab/managers/command_manager.py:302
Methodon_viewer_pause
(self, paused: bool)
src/mjlab/tasks/tracking/mdp/commands.py:555
Functionopen_stairs
( **overrides: Any, )
src/mjlab/terrains/config.py:183
Methodoutput_channels
Always ``None`` (output is flattened keypoints).
src/mjlab/rl/spatial_softmax.py:110
Methodoutput_dim
Total flattened output dimension (C * 2).
src/mjlab/rl/spatial_softmax.py:105
Functionpair_env
(device)
tests/test_domain_randomization.py:1810
Functionpair_friction
Randomize geom-pair friction overrides. Pair friction has 5 components: ``[tangent1, tangent2, spin, roll1, roll2]``. Default axis is 0 (tangent1
src/mjlab/envs/mdp/dr/pair.py:20
Methodpair_names
(self)
src/mjlab/entity/entity.py:455
Functionpd_gains
Randomize PD stiffness and damping gains. Args: env: The environment. env_ids: Environment IDs to randomize. If None, randomizes all. k
src/mjlab/envs/mdp/dr/actuator.py:30
Functionper_env_obs_func
(env)
tests/test_observation_delay.py:331
Methodper_world_default_fields
Fields with per-world defaults from mesh variant compilation.
src/mjlab/sim/sim.py:391
Functionperlin_noise
( **overrides: Any, )
src/mjlab/terrains/config.py:137
Methodphysics_dt
Physics simulation step size.
src/mjlab/envs/manager_based_rl_env.py:261
Functionpole_angle_cos_sin
Cosine and sine of the pole hinge angle. Shape: [num_envs, 2].
src/mjlab/tasks/cartpole/cartpole_env_cfg.py:83
Methodprefixed_name
The sensor name with entity prefix if applicable.
src/mjlab/sensor/builtin_sensor.py:207
Methodprev_action
Raw policy output from the previous step, before per-term scale/offset. Shape: ``(num_envs, total_action_dim)``.
src/mjlab/managers/action_manager.py:120
Methodprev_prev_action
Raw policy output from two steps ago, before per-term scale/offset. Shape: ``(num_envs, total_action_dim)``.
src/mjlab/managers/action_manager.py:126
Methodprimary_names
Primary names in the order they appear along the per-contact dim. Per-contact fields ([B, N, ...]) are laid out primary-major, so primary `pr
src/mjlab/sensor/contact_sensor.py:247
Methodprocess_actions
Process raw actions by applying scale, offset, and optional clip.
src/mjlab/envs/mdp/actions/actions.py:154
Methodprocess_actions
(self, actions: torch.Tensor)
src/mjlab/envs/mdp/actions/differential_ik.py:174
Functionprocess_docstring
Strip auto-generated dataclass docstrings (e.g. 'ClassName(*, ...)').
docs/conf.py:188
Functionprocess_signature
Suppress the ugly __init__ signature for dataclass Cfg classes.
docs/conf.py:180
Functionproject_points
r"""Projects 3D points into 2D image plane. This project 3D points into a 2D image plane. The transformation is defined by the intrinsic matr
src/mjlab/utils/lab_api/math.py:1218
Functionprojected_gravity
( env: ManagerBasedRlEnv, asset_cfg: SceneEntityCfg = _DEFAULT_ASSET_CFG, )
src/mjlab/envs/mdp/observations.py:38
Methodprojected_gravity_b
Gravity vector projected into body frame. Shape (num_envs, 3).
src/mjlab/entity/data.py:584
Functionpseudo_inertia
r"""Physics-consistent inertial randomization via the pseudo-inertia matrix. Jointly randomizes ``body_mass``, ``body_ipos``, ``body_inertia``, and
src/mjlab/envs/mdp/dr/body.py:423
Functionpush_by_setting_velocity
Push an entity by overwriting its root velocity with a sampled offset. This is an *instantaneous, mass-independent* kick: it adds a uniformly sampl
src/mjlab/envs/mdp/events.py:316
Methodqfrc_actuator
Forces produced by all actuators, mapped into joint space. For motors this is the commanded torque times the gear ratio. For position and vel
src/mjlab/entity/data.py:415
Methodqfrc_external
Forces on joints due to Cartesian wrenches applied to bodies. When a force or torque is applied to a body via ``xfrc_applied``, this property
src/mjlab/entity/data.py:426
Functionquat_apply_yaw
Rotate a vector only around the yaw-direction. Args: quat: The orientation in (w, x, y, z). Shape is (N, 4). vec: The vector in (
src/mjlab/utils/lab_api/math.py:673
Functionquat_box_plus
The box-plus operator (quaternion update) to apply an increment to a quaternion. Args: q: The initial quaternion in (w, x, y, z). Shape i
src/mjlab/utils/lab_api/math.py:608
Functionquat_env
(device)
tests/test_domain_randomization.py:500
Methodrandom
Generate a random RGBA color with specified alpha.
src/mjlab/utils/color.py:39
Functionrandom_orientation
Returns sampled rotation in 3D as quaternion. Args: num: The number of rotations to sample. device: Device to create tensor on.
src/mjlab/utils/lab_api/math.py:1299
Functionrandom_spread_boxes
( **overrides: Any, )
src/mjlab/terrains/config.py:167
Functionrandom_stairs
( **overrides: Any, )
src/mjlab/terrains/config.py:198
Functionrandom_yaw_orientation
Returns sampled rotation around z-axis. Args: num: The number of rotations to sample. device: Device to create tensor on. Re
src/mjlab/utils/lab_api/math.py:1319
Functionrandomize_terrain
Randomize the sub-terrain for each environment on reset. This picks a random terrain type (column) and difficulty level (row) for each environmen
src/mjlab/envs/mdp/events.py:58
Methodraw_action
(self)
src/mjlab/managers/action_manager.py:67
Methodraw_action
Raw actions (before scale/offset).
src/mjlab/envs/mdp/actions/actions.py:135
Methodraw_action
(self)
src/mjlab/envs/mdp/actions/differential_ik.py:171
Methodrecord_post_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:39
Methodrecord_post_step
(self)
tests/test_recorder_manager.py:42
Methodrecord_post_step
(self)
tests/test_recorder_manager.py:86
Methodrecord_post_step
(self)
tests/test_recorder_manager.py:188
Methodrecord_post_step
(self)
tests/test_recorder_manager.py:256
Methodrecord_post_step
(self)
tests/test_recorder_manager.py:475
Methodrecord_pre_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:36
Methodrecord_pre_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:83
Methodrecord_pre_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:185
Methodrecord_pre_reset
(self, env_ids: torch.Tensor)
tests/test_recorder_manager.py:471
Methodremove
(self)
tests/test_variants.py:1296
Methodrender
(self)
src/mjlab/viewer/offscreen_renderer.py:168
Methodrender
(self)
src/mjlab/envs/manager_based_rl_env.py:484
Methodrender_context
(self)
src/mjlab/sensor/sensor_context.py:97
Methodrender_mode
(self)
src/mjlab/rl/vecenv_wrapper.py:32
Methodrenderer
(self)
src/mjlab/viewer/offscreen_renderer.py:63
Functionrepeat_array_kernel
( src: wp.array(dtype=Any), # type: ignore nelems_per_world: int, dst: wp.array(dtype=Any), # type: ig
src/mjlab/sim/randomization.py:10
Functionrequires_model_fields
Mark an event function as requiring specific model fields expanded per-world. Fields listed here are registered in ``EventManager.domain_randomizat
src/mjlab/managers/event_manager.py:69
Methodreset
Reset the environment.
src/mjlab/utils/wrappers/video_recorder.py:81
Methodreset
Reset noise model state. Override in subclasses if needed.
src/mjlab/utils/noise/noise_model.py:26
Methodreset
(self, env_ids: torch.Tensor | slice | None = None)
src/mjlab/managers/curriculum_manager.py:95
Methodreset
(self, env_ids: torch.Tensor | None = None)
src/mjlab/managers/curriculum_manager.py:151
Methodreset
Resets the manager term.
src/mjlab/managers/manager_base.py:84
Methodreset
Resets the manager and returns logging info for the current step.
src/mjlab/managers/manager_base.py:119
Methodreset
(self, env_ids: torch.Tensor | None = None)
src/mjlab/managers/event_manager.py:216
Methodreset
(self, env_ids: torch.Tensor | None)
src/mjlab/managers/command_manager.py:241
Methodreset
(self, env_ids: torch.Tensor | None = None)
src/mjlab/managers/command_manager.py:320
Methodreset
Reset history buffers for specified environments. Args: env_ids: Environment indices to reset. If None, reset all environments.
src/mjlab/actuator/learned_actuator.py:122
Methodreset
Reset actuator state for specified environments. Resets delay buffers if present. Subclasses that override this should call ``super().reset(e
src/mjlab/actuator/actuator.py:365
Methodreset
(self, env_ids: torch.Tensor | slice | None = None)
src/mjlab/envs/mdp/events.py:584
Methodreset
Reset raw actions to zero for specified environments.
src/mjlab/envs/mdp/actions/actions.py:165
Methodreset
(self, env_ids: torch.Tensor | slice | None = None)
src/mjlab/envs/mdp/actions/differential_ik.py:257
Methodreset
Reset sensor state for specified environments. Invalidates the data cache. Override in subclasses that maintain internal state, but call `sup
src/mjlab/sensor/sensor.py:129
Methodreset
(self, env_ids: torch.Tensor | slice | None = None)
src/mjlab/sensor/contact_sensor.py:352
Methodreset
(self, env_ids: torch.Tensor | None = None, env=None)
tests/test_metrics_manager.py:25
Methodreset_environment
Extend BaseViewer.reset_environment to clear reward histories.
src/mjlab/viewer/native/viewer.py:445
Functionreset_joints_by_offset
( env: ManagerBasedRlEnv, env_ids: torch.Tensor | None, position_range: tuple[float, float], velocity_
src/mjlab/envs/mdp/events.py:279
← previousnext →1,701–1,800 of 2,812, ranked by callers