diff options
author | Timo Kaufmann <timokau@zoho.com> | 2019-03-27 21:27:22 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-03-27 21:27:22 +0000 |
commit | f2569b9116c39cc1ef25f9e9a65fbdf07b0e8077 (patch) | |
tree | 6cec2b053e02223383fb90ca8b1a0c7c7355d0b6 /pkgs/development/python-modules/baselines | |
parent | e7a084bdba521984d23bf6a5a18cf451033a995e (diff) | |
download | nixpkgs-f2569b9116c39cc1ef25f9e9a65fbdf07b0e8077.tar nixpkgs-f2569b9116c39cc1ef25f9e9a65fbdf07b0e8077.tar.gz nixpkgs-f2569b9116c39cc1ef25f9e9a65fbdf07b0e8077.tar.bz2 nixpkgs-f2569b9116c39cc1ef25f9e9a65fbdf07b0e8077.tar.lz nixpkgs-f2569b9116c39cc1ef25f9e9a65fbdf07b0e8077.tar.xz nixpkgs-f2569b9116c39cc1ef25f9e9a65fbdf07b0e8077.tar.zst nixpkgs-f2569b9116c39cc1ef25f9e9a65fbdf07b0e8077.zip |
python.pkgs.baselines: init at 0.1.5 (#58452)
Diffstat (limited to 'pkgs/development/python-modules/baselines')
-rw-r--r-- | pkgs/development/python-modules/baselines/default.nix | 59 | ||||
-rw-r--r-- | pkgs/development/python-modules/baselines/fix-dep-names.patch | 18 |
2 files changed, 77 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/baselines/default.nix b/pkgs/development/python-modules/baselines/default.nix new file mode 100644 index 00000000000..2fd58010f5a --- /dev/null +++ b/pkgs/development/python-modules/baselines/default.nix @@ -0,0 +1,59 @@ +{ lib +, buildPythonPackage +, fetchPypi +, pytest +, gym +, scipy +, tqdm +, joblib +, dill +, progressbar2 +, cloudpickle +, click +, pyzmq +, tensorflow +, mpi4py +}: + +buildPythonPackage rec { + pname = "baselines"; + version = "0.1.5"; + + src = fetchPypi { + inherit pname version; + sha256 = "0n1mxkcg82gzhkb4j5zzwm335r3rc1sblknqs4x6nkrrh42d65cm"; + }; + + patches = [ + # already fixed upstream + ./fix-dep-names.patch + ]; + + propagatedBuildInputs = [ + gym + scipy + tqdm + joblib + pyzmq + dill + progressbar2 + mpi4py + cloudpickle + tensorflow + click + ]; + + # fails to create a daemon, probably because of sandboxing + doCheck = false; + + checkInputs = [ + pytest + ]; + + meta = with lib; { + description = "High-quality implementations of reinforcement learning algorithms"; + homepage = https://github.com/openai/baselines; + license = licenses.mit; + maintainers = with maintainers; [ timokau ]; + }; +} diff --git a/pkgs/development/python-modules/baselines/fix-dep-names.patch b/pkgs/development/python-modules/baselines/fix-dep-names.patch new file mode 100644 index 00000000000..5b0de29247a --- /dev/null +++ b/pkgs/development/python-modules/baselines/fix-dep-names.patch @@ -0,0 +1,18 @@ +diff --git a/setup.py b/setup.py +index bf8badc..570be20 100644 +--- a/setup.py ++++ b/setup.py +@@ -10,11 +10,11 @@ setup(name='baselines', + packages=[package for package in find_packages() + if package.startswith('baselines')], + install_requires=[ +- 'gym[mujoco,atari,classic_control,robotics]', ++ 'gym', + 'scipy', + 'tqdm', + 'joblib', +- 'zmq', ++ 'pyzmq', + 'dill', + 'progressbar2', + 'mpi4py', |