MCPcopy Create free account

hub / github.com/pollen-robotics/microduck_rl / functions

Functions681 in github.com/pollen-robotics/microduck_rl

↓ 1 callersMethodtrigger_behavior
Start an episodic behavior (kick_left / kick_right / roulade). The behavior policies were trained to run from a standing start with an
scripts/infer_policy.py:731
↓ 1 callersMethodtrigger_ground_pick
Start one ground pick cycle. Automatically returns to walking when done.
scripts/infer_policy.py:683
↓ 1 callersMethodupdate_behavior
Advance the behavior timer; hand back to walking/standing when done.
scripts/infer_policy.py:782
↓ 1 callersMethodupdate_ground_pick_phase
Advance the ground pick phase; auto-exit when one full cycle completes.
scripts/infer_policy.py:716
↓ 1 callersFunctionvec
(d)
src/mjlab_microduck/tasks/mdp.py:2662
Method__call__
(self, obs)
src/mjlab_microduck/export.py:233
Method__call__
(self, obs)
src/mjlab_microduck/export.py:241
Method__enter__
(self)
scripts/infer_policy.py:163
Method__exit__
(self, *exc)
scripts/infer_policy.py:172
Method__getitem__
(self, key)
tests/test_obs_nan_guard.py:33
Method__getitem__
(self, _k)
tests/test_wheel_glide.py:38
Method__getitem__
(self, _)
tests/test_head_pose_bias.py:41
Method__getitem__
(self, _k)
tests/test_descent_speed.py:27
Method__getitem__
(self, _key)
tests/test_nan_guard.py:31
Method__init__
(self, model: mujoco.MjModel, site: int, seed: int = 0)
src/mjlab_microduck/sim/tof.py:40
Method__init__
(self, scene: Path, count: int = 1)
src/mjlab_microduck/sim/body_server.py:176
Method__init__
(self, world: World, index: int, limp: bool = False, kp: float = 200.0)
src/mjlab_microduck/sim/body_server.py:206
Method__init__
(self, cfg: "TargetAngleCommandCfg", env: ManagerBasedRlEnv)
src/mjlab_microduck/tasks/testbench_env_cfg.py:59
Method__init__
(self, cfg, env: ManagerBasedRlEnv)
src/mjlab_microduck/tasks/mdp.py:4595
Method__init__
(self, cfg, env: ManagerBasedRlEnv)
src/mjlab_microduck/tasks/mdp.py:4982
Method__init__
(self, cfg: "UniformPoseCommandCfg", env: ManagerBasedRlEnv)
src/mjlab_microduck/tasks/mdp.py:5071
Method__init__
(self, env, train_cfg: dict, log_dir=None, device="cpu", **kwargs)
src/mjlab_microduck/tasks/__init__.py:14
Method__init__
(self)
scripts/infer_policy.py:156
Method__init__
(self, model, data, walking_onnx_path=None, action_scale=1.0, bam_ctrl=None, delay_min_lag=0,
scripts/infer_policy.py:212
Method__init__
(self, onnx_path: str, action_scale: float = 1.0)
scripts/testbench_sim2real.py:75
Method__init__
(self, force=None, heights=None)
tests/test_obs_nan_guard.py:18
Method__init__
(self, data)
tests/test_obs_nan_guard.py:24
Method__init__
(self, sensors, asset)
tests/test_obs_nan_guard.py:29
Method__init__
(self, n)
tests/test_obs_nan_guard.py:38
Method__init__
(self, data)
tests/test_obs_nan_guard.py:48
Method__init__
(self, n, force)
tests/test_obs_nan_guard.py:53
Method__init__
(self, joint_pos, default_pos)
tests/test_ground_pick_pose.py:24
Method__init__
(self, names, joint_pos, default_pos)
tests/test_ground_pick_pose.py:30
Method__init__
(self, cmd)
tests/test_ground_pick_pose.py:41
Method__init__
(self, names, joint_pos, default_pos, phase)
tests/test_ground_pick_pose.py:49
Method__init__
(self, omegas)
tests/test_wheel_glide.py:17
Method__init__
(self, data)
tests/test_wheel_glide.py:23
Method__init__
(self, omegas)
tests/test_wheel_glide.py:35
Method__init__
(self, ang_vel_b=None, lin_vel_b=None, joint_pos=None, joint_vel=None)
tests/test_spin.py:90
Method__init__
(self, data, joint_ids=None)
tests/test_spin.py:100
Method__init__
(self, cmd)
tests/test_spin.py:117
Method__init__
(self, current_contact_time)
tests/test_spin.py:125
Method__init__
(self, current_contact_time)
tests/test_spin.py:130
Method__init__
(self, joint_pos, default_joint_pos)
tests/test_spin.py:369
Method__init__
(self, n)
tests/test_head_pose_bias.py:18
Method__init__
(self, data)
tests/test_head_pose_bias.py:27
Method__init__
(self, n)
tests/test_head_pose_bias.py:32
Method__init__
(self, asset, n)
tests/test_head_pose_bias.py:37
Method__init__
(self, n, dim=4)
tests/test_head_pose_bias.py:46
Method__init__
(self, n)
tests/test_head_pose_bias.py:54
Method__init__
(self, vx)
tests/test_descent_speed.py:11
Method__init__
(self, data)
tests/test_descent_speed.py:18
Method__init__
(self, vx)
tests/test_descent_speed.py:23
Method__init__
(self, n)
tests/test_nan_guard.py:13
Method__init__
(self, data)
tests/test_nan_guard.py:23
Method__init__
(self, asset)
tests/test_nan_guard.py:28
Method__init__
(self, data)
tests/test_nan_guard.py:36
Method_compute_friction_budget
( self, motor_torque: torch.Tensor, external_torque: torch.Tensor, stribeck_co
src/mjlab_microduck/actuator/friction_dr_bam.py:37
Method_debug_vis_impl
(self, visualizer: "DebugVisualizer")
src/mjlab_microduck/tasks/mdp.py:4535
Function_get_base_metadata_no_passive
(env, run_path)
src/mjlab_microduck/tasks/mdp.py:79
Function_nan_safe_reward_compute
(self, dt: float)
src/mjlab_microduck/tasks/mdp.py:32
Function_on_spark
()
tests/test_aarch64_cuda_torch.py:122
Method_reader
(self)
scripts/infer_policy.py:187
Method_resample_command
(self, env_ids: torch.Tensor)
src/mjlab_microduck/tasks/testbench_env_cfg.py:71
Method_resample_command
(self, env_ids: torch.Tensor)
src/mjlab_microduck/tasks/mdp.py:4508
Method_resample_command
(self, env_ids: torch.Tensor)
src/mjlab_microduck/tasks/mdp.py:4603
Method_resample_command
(self, env_ids: torch.Tensor)
src/mjlab_microduck/tasks/mdp.py:5010
Method_resample_command
(self, env_ids: torch.Tensor)
src/mjlab_microduck/tasks/mdp.py:5086
Function_safe_compute_returns
(self, obs)
src/mjlab_microduck/tasks/mdp.py:52
Function_soften_terrain_contacts
Soften terrain box geom contacts to reduce edge-contact NaN instability. Box terrains place adjacent geoms at different heights. The hard edges w
src/mjlab_microduck/tasks/microduck_velocity_env_cfg.py:172
Method_update_command
(self)
src/mjlab_microduck/tasks/testbench_env_cfg.py:77
Method_update_command
(self)
src/mjlab_microduck/tasks/mdp.py:4613
Method_update_command
(self)
src/mjlab_microduck/tasks/mdp.py:5013
Method_update_command
(self)
src/mjlab_microduck/tasks/mdp.py:5083
Method_update_command
(self)
src/mjlab_microduck/tasks/mdp.py:6312
Method_update_metrics
(self)
src/mjlab_microduck/tasks/testbench_env_cfg.py:80
Method_update_metrics
(self)
src/mjlab_microduck/tasks/mdp.py:4625
Method_update_metrics
(self)
src/mjlab_microduck/tasks/mdp.py:5016
Method_update_metrics
(self)
src/mjlab_microduck/tasks/mdp.py:5080
Method_update_metrics
(self)
src/mjlab_microduck/tasks/mdp.py:6315
Functionaction_over_limit_penalty
Penalise commanding a joint target beyond its hard limit (+ overshoot). Policy-side deterrent against over-driving a joint onto its mechanical st
src/mjlab_microduck/tasks/mdp.py:4909
Functionadaptive_pose_weight
Reduce pose tracking weight when robot has high velocity (recovering from push). This gives the robot freedom to deviate from the standing pose w
src/mjlab_microduck/tasks/mdp.py:3982
Functionair_time_adaptive
Air-time reward with separate swing-time windows for walking vs running. - command < command_threshold → 0 (standing, no reward) - command_t
src/mjlab_microduck/tasks/mdp.py:3799
Methodalpha
Slewed target blend: 0 = STAND target, 1 = SIT target.
src/mjlab_microduck/tasks/mdp.py:6275
Functionapply_mouth_payload_force
Hook par-step (utilisé comme reward de poids 0) : applique le POIDS de l'objet tenu dans la bouche comme force externe verticale au mouth_tip, gat
src/mjlab_microduck/tasks/mdp.py:3102
Methodas_dict
(self)
src/mjlab_microduck/publish/manifest.py:60
Functionball_forward_velocity
Ball XY velocity along the per-env kick direction, clamped to [0, max]. Dense and linear-in-speed up to ``max_speed``: every extra bit of forward
src/mjlab_microduck/tasks/mdp.py:5761
Functionball_pos_in_base
Ball position relative to the robot root, in the robot's base frame. CRITIC-ONLY observation (asymmetric actor-critic): the deployed policy has
src/mjlab_microduck/tasks/mdp.py:5827
Functionball_speed_overshoot_penalty
Ball forward speed in excess of ``target_speed`` (linear, ≥ 0). Companion to ``ball_forward_velocity`` for a target-speed kick: below the tar
src/mjlab_microduck/tasks/mdp.py:5787
Functionball_vel_in_base
Ball linear velocity in the robot's base frame. CRITIC-ONLY (see above).
src/mjlab_microduck/tasks/mdp.py:5844
Functionbam_sim
(ip)
tests/test_infer_policy_bam.py:48
Functionbase_ang_vel_imu_misaligned
base angular velocity with the SAME per-env IMU misalignment as gravity.
src/mjlab_microduck/tasks/mdp.py:3652
Functionbilateral_symmetry_penalty
L1 penalty on left/right leg asymmetry. For a bilaterally-symmetric robot the leg HOME and any symmetric target (FOLD, SIT) satisfy ``q_left
src/mjlab_microduck/tasks/mdp.py:2233
Functionbody_ang_vel_at_height
Trunk ``sum(ω_xy²)`` penalty gated by trunk z (and optionally tilt). Height-gated arrival damper: zero below ``height_low`` (ground recovery —
src/mjlab_microduck/tasks/mdp.py:763
Functionbody_impact_cost
Penalize terrain contact forces above a threshold on protected body parts. Used to discourage slamming the trunk shell or head into the ground du
src/mjlab_microduck/tasks/mdp.py:1644
Functionbody_pose_tracking_6d
Mean of 6 per-axis Gaussian rewards for tracking commanded body pose. cmd has shape (N, 6) = [x, y, z, roll, pitch, yaw] all as deltas from the
src/mjlab_microduck/tasks/mdp.py:5320
Functionbody_pose_tracking_locomotion
Locomotion-aware 6D body pose tracking. Same shape as body_pose_tracking_6d (6D cmd, mean of 6 Gaussians), but x/y/yaw are measured *relative
src/mjlab_microduck/tasks/mdp.py:5405
Functionbody_upright_gaussian
Gaussian reward on tilt magnitude — sharp pull toward fully vertical. Complements ``body_upright_linear`` (which is ``cos(tilt)`` and whose g
src/mjlab_microduck/tasks/mdp.py:710
Functionbody_upright_linear
Linear reward for body uprightness — provides gradient at every tilt angle. Returns +1 when fully upright, 0 when horizontal (prone/supine), -1 w
src/mjlab_microduck/tasks/mdp.py:683
Functionbraking_reward
Reward coming to a stop when cmd_x < 0 (brake commanded). Returns clamp(-cmd_x, 0) * exp(-fwd_vel² / vel_std²). - Silent when cmd_x ≥ 0 (coas
src/mjlab_microduck/tasks/mdp.py:1740
← previousnext →201–300 of 681, ranked by callers