diff options
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/tensorflow-tensorboard/default.nix')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/tensorflow-tensorboard/default.nix | 67 |
1 files changed, 67 insertions, 0 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/tensorflow-tensorboard/default.nix b/nixpkgs/pkgs/development/python-modules/tensorflow-tensorboard/default.nix new file mode 100644 index 000000000000..031254205066 --- /dev/null +++ b/nixpkgs/pkgs/development/python-modules/tensorflow-tensorboard/default.nix @@ -0,0 +1,67 @@ +{ lib, fetchPypi, buildPythonPackage, isPy3k +, numpy +, wheel +, werkzeug +, protobuf +, grpcio +, markdown +, absl-py +, google-auth-oauthlib +, tensorboard-plugin-wit +, tensorboard-plugin-profile +}: + +# tensorflow/tensorboard is built from a downloaded wheel, because +# https://github.com/tensorflow/tensorboard/issues/719 blocks +# buildBazelPackage. + +buildPythonPackage rec { + pname = "tensorflow-tensorboard"; + version = "2.4.0"; + format = "wheel"; + disabled = !isPy3k; + + src = fetchPypi { + pname = "tensorboard"; + inherit version format; + python = "py3"; + sha256 = "0f17h6i398n8maam0r3rssqvdqnqbwjyf96nnhf482anm1iwdq6d"; + }; + + propagatedBuildInputs = [ + numpy + werkzeug + protobuf + markdown + grpcio + absl-py + google-auth-oauthlib + tensorboard-plugin-profile + tensorboard-plugin-wit + # not declared in install_requires, but used at runtime + # https://github.com/NixOS/nixpkgs/issues/73840 + wheel + ]; + + # in the absence of a real test suite, run cli and imports + checkPhase = '' + $out/bin/tensorboard --help > /dev/null + ''; + + pythonImportsCheck = [ + "tensorboard" + "tensorboard.backend" + "tensorboard.compat" + "tensorboard.data" + "tensorboard.plugins" + "tensorboard.summary" + "tensorboard.util" + ]; + + meta = with lib; { + description = "TensorFlow's Visualization Toolkit"; + homepage = "http://tensorflow.org"; + license = licenses.asl20; + maintainers = with maintainers; [ abbradar ]; + }; +} |