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