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

   1 { lib
   2 , fetchFromGitHub
   3 , buildPythonPackage
   4 , numpy
   5 , absl-py
   6 , dm-tree
   7 , wrapt
   8 , tensorflow
   9 , tensorflow-probability
  10 , pytestCheckHook
  11 , nose }:
  12
  13 buildPythonPackage rec {
  14   pname = "trfl";
  15   version = "1.2.0";
  16
  17   src = fetchFromGitHub {
  18     owner = "deepmind";
  19     repo = pname;
  20     rev = "ed6eff5b79ed56923bcb102e152c01ea52451d4c";
  21     sha256 = "sha256-UsDUKJCHSJ4gP+P95Pm7RsPpqTJqJhrsW47C7fTZ77I=";
  22   };
  23
  24   buildInputs = [
  25     absl-py
  26     dm-tree
  27     numpy
  28     wrapt
  29   ];
  30
  31   propagatedBuildInputs = [
  32     tensorflow
  33     tensorflow-probability
  34   ];
  35
  36   checkInputs = [
  37     nose
  38     pytestCheckHook
  39   ];
  40
  41   pythonImportsCheck = [
  42     "trfl"
  43   ];
  44
  45   # Tests currently fail with assertion errors
  46   doCheck = false;
  47
  48   disabledTestPaths = [
  49     # AssertionErrors
  50     "trfl/indexing_ops_test.py"
  51     "trfl/vtrace_ops_test.py"
  52     "trfl/value_ops_test.py"
  53     "trfl/target_update_ops_test.py"
  54     "trfl/sequence_ops_test.py"
  55     "trfl/retrace_ops_test.py"
  56     "trfl/policy_ops_test.py"
  57     "trfl/policy_gradient_ops_test.py"
  58     "trfl/pixel_control_ops_test.py"
  59     "trfl/periodic_ops_test.py"
  60     "trfl/dpg_ops_test.py"
  61     "trfl/distribution_ops_test.py"
  62     "trfl/dist_value_ops_test.py"
  63     "trfl/discrete_policy_gradient_ops_test.py"
  64     "trfl/continuous_retrace_ops_test.py"
  65     "trfl/clipping_ops_test.py"
  66     "trfl/action_value_ops_test.py"
  67   ];
  68
  69   meta = with lib; {
  70     description = "TensorFlow Reinforcement Learning";
  71     homepage = "https://github.com/deepmind/trfl";
  72     license = licenses.asl20;
  73     maintainers = with maintainers; [ onny ];
  74   };
  75 }