pyrox.dev
1{
2 lib,
3 fetchPypi,
4 buildPythonPackage,
5
6 # dependencies
7 absl-py,
8 grpcio,
9 markdown,
10 numpy,
11 packaging,
12 pillow,
13 protobuf,
14 setuptools,
15 tensorboard-data-server,
16 werkzeug,
17 standard-imghdr,
18
19 versionCheckHook,
20}:
21
22buildPythonPackage rec {
23 pname = "tensorboard";
24 version = "2.20.0";
25 format = "wheel";
26
27 # tensorflow/tensorboard is built from a downloaded wheel, because
28 # https://github.com/tensorflow/tensorboard/issues/719 blocks buildBazelPackage.
29 src = fetchPypi {
30 inherit pname version format;
31 dist = "py3";
32 python = "py3";
33 hash = "sha256-ncn5eMuEwHI6z5o0XZbBhPApPRjxZruNWe4Jjmz6q6Y=";
34 };
35
36 pythonRelaxDeps = [
37 "google-auth-oauthlib"
38 "protobuf"
39 ];
40
41 dependencies = [
42 absl-py
43 grpcio
44 markdown
45 numpy
46 packaging
47 pillow
48 protobuf
49 setuptools
50 tensorboard-data-server
51 werkzeug
52
53 # Requires 'imghdr' which has been removed from python in 3.13
54 # ModuleNotFoundError: No module named 'imghdr'
55 # https://github.com/tensorflow/tensorboard/issues/6964
56 standard-imghdr
57 ];
58
59 pythonImportsCheck = [
60 "tensorboard"
61 "tensorboard.backend"
62 "tensorboard.compat"
63 "tensorboard.data"
64 "tensorboard.plugins"
65 "tensorboard.summary"
66 "tensorboard.util"
67 ];
68
69 nativeCheckInputs = [
70 versionCheckHook
71 ];
72 versionCheckProgramArg = "--version";
73
74 meta = {
75 changelog = "https://github.com/tensorflow/tensorboard/blob/${version}/RELEASE.md";
76 description = "TensorFlow's Visualization Toolkit";
77 homepage = "https://www.tensorflow.org/";
78 license = lib.licenses.asl20;
79 mainProgram = "tensorboard";
80 maintainers = [ ];
81 };
82}