pkgs/development/python-modules/baselines/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , pytest
   5 , gym
   6 , scipy
   7 , tqdm
   8 , joblib
   9 , dill
  10 , progressbar2
  11 , cloudpickle
  12 , click
  13 , pyzmq
  14 , tensorflow
  15 , mpi4py
  16 }:
  17
  18 buildPythonPackage {
  19   pname = "baselines";
  20   version = "0.1.6"; # remember to manually adjust the rev
  21   format = "setuptools";
  22
  23   src = fetchFromGitHub {
  24     owner = "openai";
  25     repo = "baselines";
  26     # Unfortunately releases are not tagged. This commit bumps the version in setup.py
  27     rev = "2bca7901f51c88cdef3ca0666c6a87c454a4dbe8";
  28     sha256 = "0j2ck7rsrcyny9qbmrw9aqvzfhv70nbign8iva2dsisa2x24gbcl";
  29   };
  30
  31   propagatedBuildInputs = [
  32     gym
  33     scipy
  34     tqdm
  35     joblib
  36     pyzmq
  37     dill
  38     progressbar2
  39     mpi4py
  40     cloudpickle
  41     tensorflow
  42     click
  43   ];
  44
  45   postPatch = ''
  46     # Needed for the atari wrapper, but the gym-atari package is not supported
  47     # in nixos anyways. Since opencv-python is not currently packaged, we
  48     # disable it.
  49     sed -ie '/opencv-python/d' setup.py
  50   '';
  51
  52   # fails to create a daemon, probably because of sandboxing
  53   doCheck = false;
  54
  55   nativeCheckInputs = [
  56     pytest
  57   ];
  58
  59   meta = with lib; {
  60     description = "High-quality implementations of reinforcement learning algorithms";
  61     homepage = "https://github.com/openai/baselines";
  62     license = licenses.mit;
  63     maintainers = with maintainers; [ timokau ];
  64   };
  65 }