tjh.dev
fork
Configure Feed
Select the types of activity you want to include in your feed.
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
fork
Configure Feed
Select the types of activity you want to include in your feed.
python3Packages.torchrl: 0.9.2 -> 0.10.0
Diff: https://github.com/pytorch/rl/compare/v0.9.2...v0.10.0
Changelog: https://github.com/pytorch/rl/releases/tag/v0.10.0
+44
-7
+44
-7
pkgs/development/python-modules/torchrl/default.nix
+44
-7
pkgs/development/python-modules/torchrl/default.nix
···
5
5
fetchFromGitHub,
6
6
7
7
# build-system
8
+
cmake,
8
9
ninja,
10
+
numpy,
11
+
pybind11,
9
12
setuptools,
10
-
which,
13
+
torch,
11
14
12
15
# dependencies
13
16
cloudpickle,
14
-
numpy,
15
17
packaging,
18
+
pyvers,
16
19
tensordict,
17
-
torch,
18
20
19
21
# optional-dependencies
20
22
# atari
21
23
gymnasium,
24
+
# brax
25
+
brax,
26
+
jax,
22
27
# checkpointing
23
28
torchsnapshot,
29
+
# dm-control
30
+
dm-control,
24
31
# gym-continuous
25
32
mujoco,
26
33
# llm
···
43
36
sentencepiece,
44
37
transformers,
45
38
vllm,
39
+
# marl
40
+
pettingzoo,
46
41
# offline-data
47
42
h5py,
48
43
huggingface-hub,
···
73
64
74
65
buildPythonPackage rec {
75
66
pname = "torchrl";
76
-
version = "0.9.2";
67
+
version = "0.10.0";
77
68
pyproject = true;
78
69
79
70
src = fetchFromGitHub {
80
71
owner = "pytorch";
81
72
repo = "rl";
82
73
tag = "v${version}";
83
-
hash = "sha256-6rU5+J70T0E7+60jihsjwlLls8jJlxKi3nmrL0xm2c0=";
74
+
hash = "sha256-DqLB1JnQ96cxVEzcXra1hFVfrN7eXTlTwPtlPClnaBA=";
84
75
};
85
76
77
+
postPatch = ''
78
+
substituteInPlace pyproject.toml \
79
+
--replace-fail "pybind11[global]" "pybind11"
80
+
'';
81
+
86
82
build-system = [
83
+
cmake
87
84
ninja
85
+
numpy
86
+
pybind11
88
87
setuptools
89
-
which
88
+
torch
90
89
];
90
+
dontUseCmakeConfigure = true;
91
91
92
92
dependencies = [
93
93
cloudpickle
94
94
numpy
95
95
packaging
96
96
tensordict
97
+
pyvers
97
98
torch
98
99
];
99
100
100
101
optional-dependencies = {
101
-
atari = gymnasium.optional-dependencies.atari;
102
+
atari = [
103
+
gymnasium
104
+
]
105
+
++ gymnasium.optional-dependencies.atari;
106
+
brax = [
107
+
brax
108
+
jax
109
+
];
102
110
checkpointing = [ torchsnapshot ];
111
+
dm-control = [ dm-control ];
103
112
gym-continuous = [
104
113
gymnasium
105
114
mujoco
···
136
109
transformers
137
110
vllm
138
111
];
112
+
marl = [
113
+
# dm-meltingpot (unpackaged)
114
+
pettingzoo
115
+
# vmas (unpackaged)
116
+
];
139
117
offline-data = [
140
118
h5py
141
119
huggingface-hub
···
152
120
torchvision
153
121
tqdm
154
122
];
123
+
open-spiel = [
124
+
# open-spiel (unpackaged)
125
+
];
155
126
rendering = [ moviepy ];
127
+
replay-buffer = [ torch ];
156
128
utils = [
157
129
git
158
130
hydra-core
131
+
# hydra-submitit-launcher (unpackaged)
159
132
tensorboard
160
133
tqdm
161
134
wandb