tjh.dev
1{ buildPythonPackage
2, lib
3, fetchFromGitHub
4, absl-py
5, enum34
6, future
7, mock
8, mpyq
9, numpy
10, portpicker
11, protobuf
12, pygame
13, s2clientprotocol
14, six
15, websocket-client
16, sc2-headless
17}:
18
19buildPythonPackage {
20 pname = "PySC2";
21 version = "1.2";
22
23 src = fetchFromGitHub {
24 owner = "deepmind";
25 repo = "pysc2";
26 rev = "39f84b01d662eb58b3d95791f59208c210afd4e7";
27 sha256 = "0dfbc2krd2rys1ji75ng2nl0ki8nhnylxljcp287bfb8qyz2m25p";
28 };
29
30 patches = [
31 ./fix-setup-for-py3.patch
32 ./parameterize-runconfig-sc2path.patch
33 ];
34
35 postPatch = ''
36 substituteInPlace "./pysc2/run_configs/platforms.py" \
37 --subst-var-by 'sc2path' '${sc2-headless}'
38 '';
39
40 propagatedBuildInputs = [
41 absl-py
42 enum34
43 future
44 mock
45 mpyq
46 numpy
47 portpicker
48 protobuf
49 pygame
50 s2clientprotocol
51 six
52 websocket-client
53 sc2-headless
54 ];
55
56 meta = {
57 description = "Starcraft II environment and library for training agents.";
58 homepage = "https://github.com/deepmind/pysc2";
59 license = lib.licenses.asl20;
60 platforms = lib.platforms.linux;
61 maintainers = with lib.maintainers; [ ];
62 };
63}