NeoRacer-RL/python.nix at main · Aeh961/NeoRacer-RL · GitHub

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
{ python }: with python; with python.pkgs;
let
  protobuf = python.pkgs.callPackage ./protoboeuf.nix { };
  mlagents_envs = buildPythonPackage rec {
    pname = "mlagents_envs";
    version = "1.1.0";
    format = "wheel";

    src = fetchPypi {
      inherit pname version;
      format = "wheel";
      dist = "py3";
      python = "py3";
      platform = "any";
      sha256 = "sha256-+7flTGMnpT8LfdYV3klBSvSW5fdmzMEPF103MbMruYU=";
    };

    propagatedBuildInputs = [
      (grpcio.override { inherit protobuf; })
      h5py
      numpy
      distutils
      protobuf
    ];

    doCheck = false;
  };
  mlagents = buildPythonPackage rec {
    pname = "mlagents";
    version = "1.1.0";
    format = "wheel";

    src = fetchPypi {
      inherit pname version;
      format = "wheel";
      dist = "py3";
      python = "py3";
      platform = "any";
      sha256 = "sha256-BnHkD7Tk2odA0Yu85H1aXvHFig250QmEb749mNcStPQ=";
    };

    propagatedBuildInputs = [
      mlagents_envs
    ];

    doCheck = false;
  };
in python.withPackages (ps: [
    matplotlib
    numpy
    nptyping
    mlagents
  ])