anchor.diff:    ];
   
     checkInputs = [
  +    distrax
       dm-haiku
       dm-sonnet
       optax
       pytestCheckHook
  +    rlax
       tensorflow-probability
       trfl
     ];
   
     pythonImportsCheck = [
       "bsuite"
  -  ];
  -
  -  disabledTestPaths = [
  -    # Disabled because tests require module rlax but this results in infinite
  -    # recursion error
  -    "bsuite/baselines/jax/actor_critic/run_test.py"
  -    "bsuite/baselines/jax/actor_critic_rnn/run_test.py"
  -    "bsuite/baselines/jax/boot_dqn/run_test.py"
  -    "bsuite/baselines/jax/dqn/run_test.py"
     ];
   
     disabledTests = [
  
positive.code: , trfl
  , optax
  , pytestCheckHook
  , dm-sonnet }:
  
  buildPythonPackage rec {
    pname = "bsuite";
    version = "0.3.5";
  
  
negative.code:     "test_episode_truncation"
    ];
  
    # escape infinite recursion with rlax
    doCheck = false;
  
    passthru.tests = {
      check = bsuite.overridePythonAttrs (_: { doCheck = true; });
    };
  
    meta = with lib; {
      description = ''
        Core RL Behaviour Suite. A collection of reinforcement learning
  