pkgs/development/python-modules/tensorflow-tensorboard/default.nix

   1 { lib, fetchPypi, buildPythonPackage, isPy3k
   2 , numpy
   3 , wheel
   4 , werkzeug
   5 , protobuf
   6 , grpcio
   7 , markdown
   8 , absl-py
   9 , google-auth-oauthlib
  10 , tensorboard-plugin-wit
  11 , tensorboard-plugin-profile
  12 }:
  13
  14 # tensorflow/tensorboard is built from a downloaded wheel, because
  15 # https://github.com/tensorflow/tensorboard/issues/719 blocks
  16 # buildBazelPackage.
  17
  18 buildPythonPackage rec {
  19   pname = "tensorflow-tensorboard";
  20   version = "2.4.0";
  21   format = "wheel";
  22   disabled = !isPy3k;
  23
  24   src = fetchPypi {
  25     pname = "tensorboard";
  26     inherit version format;
  27     python = "py3";
  28     sha256 = "0f17h6i398n8maam0r3rssqvdqnqbwjyf96nnhf482anm1iwdq6d";
  29   };
  30
  31   propagatedBuildInputs = [
  32     numpy
  33     werkzeug
  34     protobuf
  35     markdown
  36     grpcio
  37     absl-py
  38     google-auth-oauthlib
  39     tensorboard-plugin-profile
  40     tensorboard-plugin-wit
  41     # not declared in install_requires, but used at runtime
  42     # https://github.com/NixOS/nixpkgs/issues/73840
  43     wheel
  44   ];
  45
  46   # in the absence of a real test suite, run cli and imports
  47   checkPhase = ''
  48     $out/bin/tensorboard --help > /dev/null
  49   '';
  50
  51   pythonImportsCheck = [
  52     "tensorboard"
  53     "tensorboard.backend"
  54     "tensorboard.compat"
  55     "tensorboard.data"
  56     "tensorboard.plugins"
  57     "tensorboard.summary"
  58     "tensorboard.util"
  59   ];
  60
  61   meta = with lib; {
  62     description = "TensorFlow's Visualization Toolkit";
  63     homepage = "http://tensorflow.org";
  64     license = licenses.asl20;
  65     maintainers = with maintainers; [ abbradar ];
  66   };
  67 }