Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127{ lib, buildPythonPackage, fetchFromGitHub,
# build-system setuptools,
# dependencies absl-py, dm-env, dm-tree, frozendict, gym, immutabledict, matplotlib, mizani, pandas, patsy, plotnine, scikit-image, scipy, statsmodels, termcolor,
# tests distrax, dm-haiku, dm-sonnet, mock, optax, pytestCheckHook, rlax, tensorflow-probability,}:
buildPythonPackage (finalAttrs: { pname = "bsuite"; version = "0.3.6"; pyproject = true;
src = fetchFromGitHub { owner = "google-deepmind"; repo = "bsuite"; tag = finalAttrs.version; hash = "sha256-5TeNyPY0taVLhhQVXwYGMPXBeDzY+6jOt6LXBrHxE9M="; };
build-system = [ setuptools ];
dependencies = [ absl-py dm-env dm-tree frozendict gym immutabledict matplotlib mizani pandas patsy plotnine scikit-image scipy statsmodels termcolor ];
nativeCheckInputs = [ distrax dm-haiku dm-sonnet mock optax pytestCheckHook rlax tensorflow-probability ];
pythonImportsCheck = [ "bsuite" ];
disabledTestPaths = [ # Require unpackaged trfl "bsuite/baselines/tf/actor_critic/run_test.py" "bsuite/baselines/tf/actor_critic_rnn/run_test.py" ];
disabledTests = [ # ValueError: too many values to unpack (expected 4) "test_gym_cartpole"
# Tests require network connection "test_run9" "test_longer_action_sequence" "test_reset" "test_step_after_reset" "test_step_on_fresh_environment" "test_longer_action_sequence" "test_reset" "test_step_after_reset" "test_step_on_fresh_environment" "test_longer_action_sequence" "test_reset" "test_step_after_reset" "test_step_on_fresh_environment" "test_logger" "test_episode_truncation" ];
# Escape infinite recursion with rlax doCheck = false;
passthru.tests = { check = finalAttrs.finalPackage.overrideAttrs { doInstallCheck = true; }; };
meta = { description = "Collection of experiments that investigate core capabilities of a reinforcement learning (RL) agent"; homepage = "https://github.com/deepmind/bsuite"; changelog = "https://github.com/google-deepmind/bsuite/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ onny ]; };})