Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576{ lib, buildPythonPackage, fetchFromGitHub,
# build-system setuptools_80,
# dependencies click, cloudpickle, dill, gym, joblib, mpi4py, progressbar2, pyzmq, scipy, tensorflow, tqdm,}:
buildPythonPackage { pname = "baselines"; version = "0.1.6"; # remember to manually adjust the rev pyproject = true; __structuredAttrs = true;
src = fetchFromGitHub { owner = "openai"; repo = "baselines"; # Unfortunately releases are not tagged. This commit bumps the version in setup.py rev = "2bca7901f51c88cdef3ca0666c6a87c454a4dbe8"; hash = "sha256-lK1HRBdKR92E2hHZF5cFZ0P3N1aJ57pw8tazrPOZTEg="; };
build-system = [ # `pkg_resources` was removed from more recent versions # ModuleNotFoundError: No module named 'pkg_resources' setuptools_80 ];
dependencies = [ click cloudpickle dill gym joblib mpi4py progressbar2 pyzmq scipy tensorflow tqdm ];
postPatch = '' # Needed for the atari wrapper, but the gym-atari package is not supported # in nixos anyways. Since opencv-python is not currently packaged, we # disable it. sed -i -e '/opencv-python/d' setup.py '';
# fails to create a daemon, probably because of sandboxing doCheck = false;
pythonImportsCheck = [ "baselines" ];
meta = { description = "High-quality implementations of reinforcement learning algorithms"; homepage = "https://github.com/openai/baselines"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ timokau ]; };}