chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Modify MuJoCo Warp dev version check

Browse Source 
PiperOrigin-RevId: 869311325
Change-Id: I2f34bd932d7998d37b96b36e5d561f734567b8d5
This commit is contained in:
Taylor Howell
2026-02-12 11:38:34 -08:00
committed by Copybara-Service
parent 716ae107c6
commit 881544c0c5
1 changed files with 7 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/mujoco/mjx/third_party/mujoco_warp/_src/io.py
Vendored
+7 -13
View File
@@ -15,12 +15,12 @@
import dataclasses import dataclasses
import importlib.metadata import importlib.metadata
import re
import warnings import warnings
from typing import Any, Optional, Sequence from typing import Any, Optional, Sequence
import mujoco import mujoco
import numpy as np import numpy as np
import packaging.version
import warp as wp import warp as wp
from mujoco.mjx.third_party.mujoco_warp._src import bvh from mujoco.mjx.third_party.mujoco_warp._src import bvh
@@ -31,18 +31,12 @@ from mujoco.mjx.third_party.mujoco_warp._src import warp_util
def _is_mujoco_dev() -> bool: def _is_mujoco_dev() -> bool:
_DEV_VERSION_PATTERN = re.compile(r"^\d+\.\d+\.\d+.+") # anything after x.y.z """Checks if mujoco version is > 3.4.0."""
version_str = getattr(mujoco, "__version__", None)
version = getattr(__import__("mujoco"), "__version__", None) if not version_str:
if version and _DEV_VERSION_PATTERN.match(version): version_str = importlib.metadata.version("mujoco")
return True version_str = version_str.split("-")[0].split(".dev")[0]
return packaging.version.parse(version_str) > packaging.version.parse("3.4.0")
# fall back to metadata
dist_version = importlib.metadata.version("mujoco")
if _DEV_VERSION_PATTERN.match(dist_version):
return True
return False
BLEEDING_EDGE_MUJOCO = _is_mujoco_dev() BLEEDING_EDGE_MUJOCO = _is_mujoco_dev()