1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
{
lib,
stdenv,
buildPythonPackage,
fetchFromGitHub,
# build-system
setuptools,
# dependencies
cloudpickle,
farama-notifications,
numpy,
typing-extensions,
# optional-dependencies
# atari
ale-py,
# tests
array-api-compat,
dill,
flax,
jax,
jaxlib,
matplotlib,
mujoco,
moviepy,
opencv4,
pybox2d,
pygame,
pytestCheckHook,
scipy,
torch,
}:
buildPythonPackage rec {
pname = "gymnasium";
version = "1.2.3";
pyproject = true;
src = fetchFromGitHub {
owner = "Farama-Foundation";
repo = "gymnasium";
tag = "v${version}";
hash = "sha256-b712BPs7AS8UE8Zsu9GW/J6Vag9NB/x728MtQ5yrjbY=";
};
build-system = [ setuptools ];
dependencies = [
cloudpickle
farama-notifications
numpy
typing-extensions
];
optional-dependencies = {
atari = [
ale-py
];
};
pythonImportsCheck = [ "gymnasium" ];
nativeCheckInputs = [
array-api-compat
dill
flax
jax
jaxlib
matplotlib
moviepy
mujoco
opencv4
pybox2d
pygame
pytestCheckHook
scipy
torch
];
# if `doCheck = true` on Darwin, `jaxlib` is evaluated, which is both
# marked as broken and throws an error during evaluation if the package is evaluated anyway.
# disabling checks on Darwin avoids this and allows the package to be built.
# if jaxlib is ever fixed on Darwin, remove this.
doCheck = !stdenv.hostPlatform.isDarwin;
disabledTestPaths = [
# Unpackaged `mujoco-py` (Openai's mujoco) is required for these tests.
"tests/envs/mujoco/test_mujoco_custom_env.py"
"tests/envs/mujoco/test_mujoco_rendering.py"
"tests/envs/mujoco/test_mujoco_v5.py"
# Rendering tests failing in the sandbox
"tests/wrappers/vector/test_human_rendering.py"
# These tests need to write on the filesystem which cause them to fail.
"tests/utils/test_save_video.py"
"tests/wrappers/test_record_video.py"
];
preCheck = ''
export SDL_VIDEODRIVER=dummy
'';
disabledTests = [
# Succeeds for most environments but `test_render_modes[Reacher-v4]` fails because it requires
# OpenGL access which is not possible inside the sandbox.
"test_render_mode"
];
meta = {
description = "Standard API for reinforcement learning and a diverse set of reference environments (formerly Gym)";
homepage = "https://github.com/Farama-Foundation/Gymnasium";
changelog = "https://github.com/Farama-Foundation/Gymnasium/releases/tag/v${version}";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ GaetanLepage ];
};
}
|
