MCPcopy Create free account

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

Functions681 in github.com/pollen-robotics/microduck_rl

↓ 44 callersMethodget_command
(self, data)
src/mjlab_microduck/actuator/friction_dr_bam.py:101
↓ 36 callersFunctionmake_microduck_roller_standup_env_cfg
Env « se relever sur rollers » : départ au sol, cible = debout sur roues.
src/mjlab_microduck/tasks/microduck_roller_standup_env_cfg.py:135
↓ 14 callersFunction_servo_joint_pos
(env: "ManagerBasedRlEnv", asset: Entity)
src/mjlab_microduck/tasks/mdp.py:146
↓ 14 callersMethodfind_joints
(self, pattern)
tests/test_spin.py:104
↓ 13 callersFunctionmake_microduck_roller_slope_env_cfg
(play: bool = False)
src/mjlab_microduck/tasks/microduck_roller_slope_env_cfg.py:66
↓ 11 callersFunction_phase_cmd
Commande du slot telle que la voit la policy : [cos(2*pi*phi), sin(...), 0].
tests/test_spin.py:141
↓ 11 callersFunction_servo_default_joint_pos
(env: "ManagerBasedRlEnv", asset: Entity)
src/mjlab_microduck/tasks/mdp.py:154
↓ 10 callersFunctionmake_microduck_spin_env_cfg
Env spin sur rollers, piloté par la phase du slot ground-pick.
src/mjlab_microduck/tasks/microduck_spin_env_cfg.py:87
↓ 9 callersMethod_update_command
Update self.command (fed into obs) based on current policy and commands. Legacy mode (new_cmd_obs=False): self.command is 3D. New mod
scripts/infer_policy.py:460
↓ 8 callersFunction_update_roulade_accum
Integrate forward pitch rate into the per-env rotation accumulator. Step-guarded so that multiple reward terms reading the accumulator in the
src/mjlab_microduck/tasks/mdp.py:6720
↓ 6 callersFunction_gp_phase
(env: ManagerBasedRlEnv, command_name: str)
src/mjlab_microduck/tasks/mdp.py:2982
↓ 6 callersFunction_roulade_completion_gate
Smoothstep on the progress frontier: 0 below gate_lo rad, 1 above gate_hi. State-based replacement for the old phase-clock landing window — it ca
src/mjlab_microduck/tasks/mdp.py:6763
↓ 6 callersFunction_servo_joint_ids
Entity-local indices of the servo (non-``passive_``) joints, cached. All joint-index-based reward/event params in this module (``joint_indices``,
src/mjlab_microduck/tasks/mdp.py:126
↓ 6 callersFunctionmake_microduck_ground_pick_env_cfg
Create Microduck ground pick environment configuration.
src/mjlab_microduck/tasks/microduck_ground_pick_env_cfg.py:114
↓ 6 callersFunctionmake_microduck_velocity_rollers_env_cfg
Create Microduck roller skate velocity tracking environment configuration.
src/mjlab_microduck/tasks/microduck_velocity_rollers_env_cfg.py:85
↓ 6 callersFunctionramp_angle_by_difficulty
Angle de rampe (radians) interpolé linéairement par la difficulté [0,1].
src/mjlab_microduck/tasks/slope_terrain.py:26
↓ 5 callersFunction_empty_terrain_spec
()
tests/test_slope_terrain.py:29
↓ 5 callersFunction_fallen_mask
Per-env float mask: 1.0 where the robot counts as FALLEN — trunk height below `gate_z_below` OR tilt beyond `gate_tilt_above_deg`. Used to gate th
src/mjlab_microduck/tasks/mdp.py:507
↓ 5 callersFunction_roulade_state
(env: ManagerBasedRlEnv)
src/mjlab_microduck/tasks/mdp.py:6709
↓ 5 callersFunction_run
(env, steps, **kw)
tests/test_head_pose_bias.py:80
↓ 5 callersFunction_set_pose
(env, z, pitch_deg)
tests/test_head_pose_bias.py:73
↓ 5 callersMethodfunction
( self, difficulty: float, spec: mujoco.MjSpec, rng )
src/mjlab_microduck/tasks/slope_terrain.py:55
↓ 5 callersFunctionmake_microduck_roller_crouch_env_cfg
Env crouch-glide sur rollers, piloté par la phase du slot ground-pick.
src/mjlab_microduck/tasks/microduck_roller_crouch_env_cfg.py:116
↓ 5 callersFunctionmake_microduck_velocity_env_cfg
Create Microduck velocity tracking environment configuration.
src/mjlab_microduck/tasks/microduck_velocity_env_cfg.py:193
↓ 5 callersFunctionphase_pose_blend
Blend 0..1 le long de la phase [0,1) — 0 = pose STAND, 1 = pose DOWN. [0, descent_end) : 0 -> 1 (se baisser) [descent_end, hold_end):
src/mjlab_microduck/tasks/mdp.py:2577
↓ 5 callersFunctionslope_move_masks
Masques de promotion/rétrogradation du curriculum de pente. move_up : a parcouru plus de 40% de la tuile → il a dévalé la rampe,
src/mjlab_microduck/tasks/mdp.py:3504
↓ 4 callersFunction_force
(n, bad_env=None, value=float("nan"))
tests/test_obs_nan_guard.py:60
↓ 4 callersFunction_posture_height
(slewed target trunk z, actual trunk z) per env.
src/mjlab_microduck/tasks/mdp.py:6368
↓ 4 callersFunction_tiny_policy
A one-layer 'policy' with the daemon's shape, so the ONNX checks run without torch.
tests/test_publish_manifest.py:66
↓ 4 callersFunctiondescent_speed_reward
Récompense la vitesse d'avance vers le BAS de la pente (monde +x). La rampe descend en +x, donc la vitesse linéaire monde en x mesure la prog
src/mjlab_microduck/tasks/mdp.py:1035
↓ 4 callersMethodreset
(self)
scripts/testbench_sim2real.py:84
↓ 4 callersFunctionrobot_state_is_nan
Terminate environments where MuJoCo produced NaN joint positions. MuJoCo's contact solver can overflow to NaN under extreme penetration or im
src/mjlab_microduck/tasks/mdp.py:963
↓ 4 callersFunctionwheel_glide_reward
Récompense le ROULEMENT des roues vers l'avant (glisse), plafonné. Contrairement à descent_speed (vitesse de la BASE, qu'on peut atteindre en
src/mjlab_microduck/tasks/mdp.py:1093
↓ 3 callersMethod__init__
(self, cfg, env: ManagerBasedRlEnv)
src/mjlab_microduck/tasks/mdp.py:6261
↓ 3 callersMethod_apply_torque
Torque off means limp, not frozen. Refusing to command a fallen robot only freezes it in the pose it fell in — which is why `RobotIo:
src/mjlab_microduck/sim/body_server.py:372
↓ 3 callersFunction_ball_kick_dir
Per-env world-frame kick direction (XY unit vector), lazily allocated. Set by ``reset_ball_in_front_of_foot`` to the robot's forward direction at
src/mjlab_microduck/tasks/mdp.py:5697
↓ 3 callersFunction_env
(cur, phase)
tests/test_ground_pick_pose.py:64
↓ 3 callersFunction_finite
(x: torch.Tensor)
src/mjlab_microduck/tasks/mdp.py:5210
↓ 3 callersFunction_forward_progress_gate
0→1 ramp in body forward speed: 0 when standing still, 1 at/above v_ref. Used to gate stride-shaping rewards so that stepping which does NOT prop
src/mjlab_microduck/tasks/mdp.py:4688
↓ 3 callersFunction_packages
(name)
tests/test_aarch64_cuda_torch.py:32
↓ 3 callersFunction_posture_blend
Target blend ∈ [0, 1] (0 = STAND, 1 = SIT) for the posture rewards. Uses the SitStandCommand's slewed ``alpha`` (the moving setpoint) when the
src/mjlab_microduck/tasks/mdp.py:6334
↓ 3 callersFunction_posture_targets
(target blend, per-env joint target) for the commanded posture. STAND target = default_joint_pos (HOME); SIT target = HOME with the keyframe
src/mjlab_microduck/tasks/mdp.py:6347
↓ 3 callersFunction_servo_joint_vel
(env: "ManagerBasedRlEnv", asset: Entity)
src/mjlab_microduck/tasks/mdp.py:150
↓ 3 callersFunction_spin_gate
( env: ManagerBasedRlEnv, command_name: str, rate_max: float, accel_end: float, hold_end:
src/mjlab_microduck/tasks/mdp.py:5927
↓ 3 callersFunction_wheel_env
(vel_rows, phases)
tests/test_spin.py:258
↓ 3 callersFunctionmake_microduck_standup_env_cfg
Create Microduck stand environment configuration (sit-keyframe start).
src/mjlab_microduck/tasks/microduck_standup_env_cfg.py:152
↓ 3 callersFunctionmake_target_schedule
Return one target angle per control step.
scripts/testbench_sim2real.py:55
↓ 3 callersFunctionphase_pose_track
Gaussienne sur la pose articulaire vs cible interpolée STAND<->DOWN. Reward directif : indique la config articulaire exacte à chaque phase. Se
src/mjlab_microduck/tasks/mdp.py:2817
↓ 3 callersMethodquat_rotate_inverse
Rotate a vector by the inverse of a quaternion [w, x, y, z].
scripts/infer_policy.py:611
↓ 3 callersMethodset_vel_cmd
Set velocity command (used for walking / policy switching).
scripts/infer_policy.py:535
↓ 3 callersFunctionspin_phase_from_command
Récupère la phase [0,1) depuis la commande [cos(2πφ), sin(2πφ), 0] du slot.
src/mjlab_microduck/tasks/mdp.py:5910
↓ 3 callersMethodstep
(self, q: float, qd: float, target: float)
scripts/testbench_sim2real.py:87
↓ 2 callersFunction_backlash_encoder_ids
(main_ids, backlash_ids, mask) — cached per (entity, joint selection). mask is 1.0 where a matching passive_<name>_backlash joint exists, so the
src/mjlab_microduck/tasks/mdp.py:6145
↓ 2 callersFunction_crouch_pose_error
(cur, target) joint tensors for the phase-interpolated crouch pose. Target interpolates per joint STAND <-> crouch_pose by the 4-segment blend
src/mjlab_microduck/tasks/mdp.py:1296
↓ 2 callersFunction_fail
(msg: str)
src/mjlab_microduck/publish/cli.py:90
↓ 2 callersFunction_head_top_down
True where the head-top axis points at the floor (dot with -z > min).
src/mjlab_microduck/tasks/mdp.py:6687
↓ 2 callersFunction_imu_misalignment_quat
Per-env constant IMU mounting-misalignment rotation (sampled once). Models a fixed small mounting/calibration error of the IMU on each robot.
src/mjlab_microduck/tasks/mdp.py:3618
↓ 2 callersFunction_kick_pose_error
(cur, target) pour le geste de shoot, joints résolus PAR NOM. Les 3 poses partagent les mêmes clés (14 joints). L'ordre des noms est donné pa
src/mjlab_microduck/tasks/mdp.py:2637
↓ 2 callersFunction_lateral_axis_z
World-z component of the body's lateral (y) axis. 0 = flat/sagittal.
src/mjlab_microduck/tasks/mdp.py:6682
↓ 2 callersFunction_leg_env
(pos_rows, phases)
tests/test_spin.py:330
↓ 2 callersFunction_markers
(pkg)
tests/test_aarch64_cuda_torch.py:41
↓ 2 callersFunction_multistage_target_height
Same logic as _multistage_target_pose but for trunk z height. waypoints: [{"frac": float, "height": float}, ...].
src/mjlab_microduck/tasks/mdp.py:2304
↓ 2 callersFunction_multistage_target_pose
Compute the time-interpolated joint target across N waypoints. waypoints: ordered list of dicts {"frac": float in [0,1],
src/mjlab_microduck/tasks/mdp.py:2262
↓ 2 callersFunction_phase_pose_error
(cur, target) pour la pose interpolée par la phase, résolue PAR NOM. Cible = source + blend(phase)·(target_pose - source), source = STAND (`s
src/mjlab_microduck/tasks/mdp.py:2775
↓ 2 callersMethod_print_body_cmd
(self)
scripts/infer_policy.py:582
↓ 2 callersFunction_registry
(pkg)
tests/test_aarch64_cuda_torch.py:37
↓ 2 callersMethod_resample_command
(self, env_ids: torch.Tensor)
src/mjlab_microduck/tasks/mdp.py:6279
↓ 2 callersFunction_set_arm_mass
(spec: mujoco.MjSpec, mass: float)
src/mjlab_microduck/robot/testbench_constants.py:24
↓ 2 callersFunction_spin_target_rate
( env: ManagerBasedRlEnv, command_name: str, rate_max: float, accel_end: float, hold_end:
src/mjlab_microduck/tasks/mdp.py:5915
↓ 2 callersMethodapply_action
Apply action to MuJoCo controls with optional delay.
scripts/infer_policy.py:865
↓ 2 callersFunctioncrouch_pose_blend
Blend 0..1 le long de la phase [0,1) — 0 = pose debout, 1 = pose accroupie. [0, descent_end) : 0 -> 1 (se baisser) [descent_end, hold_e
src/mjlab_microduck/tasks/mdp.py:1273
↓ 2 callersFunctionduck_prefix
`""` for the first duck — it is the scene's own — and `d1_`, `d2_` … for attached ones.
src/mjlab_microduck/sim/body_server.py:105
↓ 2 callersMethodget_observations
Collect observations matching policy input. Order for velocity/standing task: 1. base_ang_vel (3D) 2. raw_accelerometer OR pr
scripts/infer_policy.py:655
↓ 2 callersFunctionget_walk_rollers_spec
()
src/mjlab_microduck/robot/microduck_constants.py:59
↓ 2 callersFunctioninspect_onnx
The graph's single input and output widths, as the daemon checks them at load.
src/mjlab_microduck/publish/manifest.py:233
↓ 2 callersFunctionmain
()
src/mjlab_microduck/export.py:280
↓ 2 callersFunctionmake_microduck_velocity_swizzle_env_cfg
Roller swizzle env: the stride env minus its anti-swizzle terms, plus symmetry and grounded rewards. Everything else (robot, obs, command, DR) is
src/mjlab_microduck/tasks/microduck_velocity_swizzle_env_cfg.py:34
↓ 2 callersFunctionmake_microduck_velstand_env_cfg
(play: bool = False, rough: bool = False)
src/mjlab_microduck/tasks/microduck_velstand_env_cfg.py:162
↓ 2 callersFunctionphase_pose_track_l1
Bootstrap L1 vers la cible interpolée (pénalité négative). Gradient constant partout — donne une direction vers la cible même quand la gaussi
src/mjlab_microduck/tasks/mdp.py:2841
↓ 2 callersFunctionphase_rise_gate
Gate montante pour le RETOUR : 0 avant hold_end, 0->1 sur [hold_end, rise_end), 1 après (repos debout).
src/mjlab_microduck/tasks/mdp.py:2970
↓ 2 callersFunctionrun
(cfg: PublishConfig)
src/mjlab_microduck/publish/cli.py:143
↓ 2 callersFunctionrun_export
(task_id: str, cfg: ExportConfig)
src/mjlab_microduck/export.py:72
↓ 2 callersMethodset_position_targets
Send joint position targets to the actuators. BAM: the firmware position loop lives in the controller (ctrl is the motor TORQUE it wr
scripts/infer_policy.py:884
↓ 2 callersFunctionspin_rate_by_phase
Vitesse de lacet cible (rad/s, positive = anti-horaire) le long de la phase.
src/mjlab_microduck/tasks/mdp.py:5874
↓ 2 callersFunctionsubmit
Parse submission args from ``argv`` and launch the HF job.
src/mjlab_microduck/hf_jobs.py:241
↓ 1 callersMethod__init__
(self, entity, cmd=None, sensors=None)
tests/test_spin.py:135
↓ 1 callersFunction_aarch64_entry
The entry whose resolution-markers SELECT linux-aarch64.
tests/test_aarch64_cuda_torch.py:45
↓ 1 callersMethod_alpha_from_height
(self)
src/mjlab_microduck/tasks/mdp.py:6287
↓ 1 callersFunction_await_scheduling
Poll until the job leaves SCHEDULING (image pull / queue / mounts). This phase legitimately takes minutes (the pytorch image pull alone is ~5
src/mjlab_microduck/hf_jobs.py:214
↓ 1 callersMethod_body_idx
Map an axis name to the body_cmd index, depending on the active mode.
scripts/infer_policy.py:598
↓ 1 callersFunction_build_tarball
Create a tarball of HEAD + uncommitted tracked changes. Returns short SHA.
src/mjlab_microduck/hf_jobs.py:151
↓ 1 callersFunction_default_name
(repo: str)
src/mjlab_microduck/publish/cli.py:138
↓ 1 callersMethod_end_behavior
(self)
scripts/infer_policy.py:790
↓ 1 callersMethod_end_ground_pick
Switch back after a ground pick cycle completes.
scripts/infer_policy.py:703
↓ 1 callersFunction_format_duration
(seconds: float)
scripts/wandb_utils.py:48
↓ 1 callersFunction_get_tensors
( device: torch.device, )
src/mjlab_microduck/tasks/symmetry.py:101
↓ 1 callersFunction_invoked_as_train
True when argv[0] is mjlab's trainer (console script or `-m`). `play --hf-jobs` must NOT submit a training job; let that command's own parser
src/mjlab_microduck/train_hook.py:37
↓ 1 callersFunction_iteration_of
(checkpoint_path: Path | None)
src/mjlab_microduck/export.py:65
↓ 1 callersFunction_mae
(a: np.ndarray, b: np.ndarray)
scripts/testbench_sim2real.py:421
↓ 1 callersFunction_now_utc
()
src/mjlab_microduck/publish/manifest.py:64
next →1–100 of 681, ranked by callers