35 lines
1.5 KiB
Bash
35 lines
1.5 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
# Generate type stubs for MuJoCo using pybind11-stubgen.
|
||
|
|
# This helps suppress pyright/ty errors for MuJoCo's C++ bindings.
|
||
|
|
#
|
||
|
|
# The output is post-processed by postprocess_stubs.py to make it
|
||
|
|
# platform-independent (it drops the OpenGL backend stubs, which differ between
|
||
|
|
# macOS and Linux). This keeps the committed stubs identical on every platform,
|
||
|
|
# so CI can regenerate them and fail if they drift from the installed mujoco
|
||
|
|
# version. pybind11-stubgen is pinned so its output is reproducible.
|
||
|
|
|
||
|
|
set -e
|
||
|
|
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
|
|
||
|
|
# --no-sync runs in the already-synced project environment (so mujoco is
|
||
|
|
# importable) without disturbing it; the conflicting cpu/cu128 extras mean a
|
||
|
|
# plain "uv run" would re-sync and churn the environment.
|
||
|
|
STUBGEN_VERSION="2.5.5"
|
||
|
|
STUBGEN=(uv run --no-sync --with "pybind11-stubgen==${STUBGEN_VERSION}" pybind11-stubgen)
|
||
|
|
|
||
|
|
echo "Generating MuJoCo type stubs..."
|
||
|
|
"${STUBGEN[@]}" mujoco -o "$SCRIPT_DIR" --ignore-all-errors
|
||
|
|
|
||
|
|
# mujoco.viewer pulls in GLFW, which may not import on a headless machine.
|
||
|
|
# Skip it in that case rather than failing; the committed viewer stub is kept.
|
||
|
|
if ! "${STUBGEN[@]}" mujoco.viewer -o "$SCRIPT_DIR" --ignore-all-errors; then
|
||
|
|
echo "Warning: could not generate mujoco.viewer stubs (no GLFW?). Skipping."
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "Post-processing stubs to make them platform-independent..."
|
||
|
|
uv run --no-sync python "$SCRIPT_DIR/postprocess_stubs.py"
|
||
|
|
|
||
|
|
echo "MuJoCo stubs generated successfully in $SCRIPT_DIR/mujoco/"
|