depot/third_party/nixpkgs/pkgs/development/python-modules/tensorboard/default.nix
Default email bb584b27e9 Project import generated by Copybara.
GitOrigin-RevId: 5181d5945eda382ff6a9ca3e072ed6ea9b547fee
2022-04-15 03:41:22 +02:00

90 lines
2 KiB
Nix

{ lib
, fetchPypi
, buildPythonPackage
, pythonOlder
, pythonAtLeast
, numpy
, wheel
, werkzeug
, protobuf
, grpcio
, markdown
, absl-py
, google-auth-oauthlib
, setuptools
, tensorboard-data-server
, 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 = "tensorboard";
version = "2.8.0";
format = "wheel";
disabled = pythonOlder "3.6" || pythonAtLeast "3.11";
src = fetchPypi {
inherit pname version format;
dist = "py3";
python = "py3";
hash = "sha256-ZaM45EJOkHnyYEkjvb4wF5KtzirOG+aNprPd8AUXDe8=";
};
postPatch = ''
chmod u+rwx -R ./dist
pushd dist
wheel unpack --dest unpacked ./*.whl
pushd unpacked/tensorboard-${version}
substituteInPlace tensorboard-${version}.dist-info/METADATA \
--replace "google-auth (<2,>=1.6.3)" "google-auth (<3,>=1.6.3)" \
--replace "google-auth-oauthlib (<0.5,>=0.4.1)" "google-auth-oauthlib (<0.6,>=0.4.1)"
popd
wheel pack ./unpacked/tensorboard-${version}
popd
'';
propagatedBuildInputs = [
absl-py
grpcio
google-auth-oauthlib
markdown
numpy
protobuf
setuptools
tensorboard-data-server
tensorboard-plugin-profile
tensorboard-plugin-wit
werkzeug
# 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 = "https://www.tensorflow.org/";
license = licenses.asl20;
maintainers = with maintainers; [ abbradar ];
};
}