lol

python3Packages.brax: 0.12.3 -> 0.12.4 (#416535)

authored by

Nick Cao and committed by
GitHub
ed130eb8 9bb92e23

+11 -24
+11 -24
pkgs/development/python-modules/brax/default.nix
··· 9 9 10 10 # dependencies 11 11 absl-py, 12 - dm-env, 13 12 etils, 14 13 flask, 15 14 flask-cors, 16 15 flax, 17 - grpcio, 18 - gym, 19 16 jax, 20 17 jaxlib, 21 18 jaxopt, ··· 27 24 optax, 28 25 orbax-checkpoint, 29 26 pillow, 30 - pytinyrenderer, 31 27 scipy, 32 28 tensorboardx, 33 29 typing-extensions, 34 30 35 31 # tests 32 + dm-env, 33 + gym, 36 34 pytestCheckHook, 37 35 pytest-xdist, 38 36 transforms3d, ··· 40 38 41 39 buildPythonPackage rec { 42 40 pname = "brax"; 43 - version = "0.12.3"; 41 + version = "0.12.4"; 44 42 pyproject = true; 45 43 46 44 src = fetchFromGitHub { 47 45 owner = "google"; 48 46 repo = "brax"; 49 47 tag = "v${version}"; 50 - hash = "sha256-WshTiWK6XpwK2h/aw/YogA5pGo5U7RdZBz6UjD1Ft/4="; 48 + hash = "sha256-/eb0WjMzHwD1tjTyZ2fb2dzvGrWnyOLcVLOx4BeKvqk="; 51 49 }; 52 50 53 51 build-system = [ ··· 56 54 57 55 dependencies = [ 58 56 absl-py 59 - dm-env 60 57 etils 61 58 flask 62 59 flask-cors 63 60 flax 64 - grpcio 65 - gym 66 61 jax 67 62 jaxlib 68 63 jaxopt ··· 74 69 optax 75 70 orbax-checkpoint 76 71 pillow 77 - pytinyrenderer 78 72 scipy 79 73 tensorboardx 80 74 typing-extensions 81 75 ]; 82 76 83 77 nativeCheckInputs = [ 78 + dm-env 79 + gym 84 80 pytestCheckHook 85 81 pytest-xdist 86 82 transforms3d 87 83 ]; 88 84 89 - disabledTests = 90 - [ 91 - # Broken after mujoco was updated to 3.3.3: 92 - # TypeError: Data.init() got an unexpected keyword argument 'contact' 93 - # Reported upstream: https://github.com/google/brax/issues/618 94 - "test_pendulum" 95 - "test_pipeline_ant" 96 - "test_pipeline_init_with_ctrl" 97 - ] 98 - ++ lib.optionals stdenv.hostPlatform.isAarch64 [ 99 - # Flaky: 100 - # AssertionError: Array(-0.00135638, dtype=float32) != 0.0 within 0.001 delta (Array(0.00135638, dtype=float32) difference) 101 - "test_pendulum_period2" 102 - ]; 85 + disabledTests = lib.optionals stdenv.hostPlatform.isAarch64 [ 86 + # Flaky: 87 + # AssertionError: Array(-0.00135638, dtype=float32) != 0.0 within 0.001 delta (Array(0.00135638, dtype=float32) difference) 88 + "test_pendulum_period2" 89 + ]; 103 90 104 91 disabledTestPaths = [ 105 92 # ValueError: matmul: Input operand 1 has a mismatch in its core dimension