pyrox.dev
lol
python3Packages.brax: 0.12.3 -> 0.12.4 (#416535)
+11
-24
+11
-24
pkgs/development/python-modules/brax/default.nix
+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