Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273{ lib, buildPythonPackage, mujoco, fetchFromGitHub,
# build-system setuptools,
# dependencies absl-py, etils, numpy, warp-lang,
# optional-dependencies jax,
# tests pytestCheckHook, writableTmpDirAsHomeHook,}:
buildPythonPackage (finalAttrs: { pname = "mujoco-warp"; inherit (mujoco) version; pyproject = true; __structuredAttrs = true;
src = fetchFromGitHub { owner = "google-deepmind"; repo = "mujoco_warp"; tag = "v${finalAttrs.version}"; hash = "sha256-uk/h7Rl5jrR8fPsYvJ6zy4+y9vMTpQ0mmCQys9eTUg8="; };
build-system = [ setuptools ];
dependencies = [ absl-py etils mujoco numpy warp-lang ];
optional-dependencies = { cpu = [ jax ]; cuda = [ jax ]; };
nativeCheckInputs = [ pytestCheckHook writableTmpDirAsHomeHook ];
pythonImportsCheck = [ "mujoco_warp" ];
meta = { description = "GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware"; homepage = "https://github.com/google-deepmind/mujoco_warp"; changelog = "https://github.com/google-deepmind/mujoco_warp/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ nim65s ]; };})