gnn/pyproject.toml at main · tensorflow/gnn · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
[project]
name = "tensorflow_gnn"
description = "A library for building scalable graph neural networks in TensorFlow."
# TODO(b/316329189): The relative links from README.md don't work on PyPI.
readme = "README.md"
keywords = ["tensorflow", "gnn", "graph"]
dynamic = ["version"]
license = "Apache-2.0"
requires-python = ">=3.10, <4"
dependencies = [
  "ml-collections>=1.1.0",
  "networkx>=3.4.2",
  "pyarrow>=18.1.0",
  "tensorflow>=2.17.0, <3",
  # Raised lower bound to 2.57 to enable python 3.12+ support.
  "apache-beam>=2.57, <3",
  # This dependency comes from apache beam implementing it as an extra for 2.70.
  # To maintain current compatibility with pre and post 2.70, we need to include it here. b/465442394
  "crcmod>=1.7, <2",
  "tensorboard",
  "tf-keras>=2.17.0, <3",
]
authors = [
  {name = "Google LLC", email = "tensorflow-gnn@googlegroups.com"},
]
classifiers = [
  "Development Status :: 5 - Production/Stable",
  "Intended Audience :: Developers",
  "Intended Audience :: Science/Research",
  "Operating System :: MacOS :: MacOS X",
  "Operating System :: POSIX :: Linux",
  "Operating System :: Microsoft :: Windows",
  "Programming Language :: Python",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3 :: Only",
  "Topic :: Scientific/Engineering",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
  "Topic :: Scientific/Engineering :: Mathematics",
  "Topic :: Software Development",
  "Topic :: Software Development :: Libraries",
  "Topic :: Software Development :: Libraries :: Python Modules",
]

[project.urls]
homepage = "https://github.com/tensorflow/gnn"
source = "https://github.com/tensorflow/gnn.git"

[project.optional-dependencies]
vizier = [
  "google-vizier>=0.0.13, !=0.1.23",  # Moved to extras due to dependency conflicts. b/394062744
  "tensorflow>=2.18.0, <3",
]

[dependency-groups]
test-base = [
  "mock",
]
test-released = [
  "ai-edge-litert",
  {include-group = "test-base"},
]
test-nightly = [
  "tf-nightly",
  "tf-keras-nightly",
  "ai-edge-litert-nightly",
  {include-group = "test-base"},
]

[project.scripts]
tfgnn_convert_ogb_dataset = 'tensorflow_gnn.converters.ogb.convert_ogb_dataset:main'
tfgnn_sampler = 'tensorflow_gnn.experimental.sampler.beam.sampler:main'
# copybara:comment_begin(NetworkX utils)
tfgnn_nx_converter = 'tensorflow_gnn.sampler.nx_converter:main'
tfgnn_nx_generator = 'tensorflow_gnn.sampler.nx_generator:main'
# copybara:comment_end
tfgnn_generate_training_data = 'tensorflow_gnn.tools.generate_training_data:main'
tfgnn_print_training_data = 'tensorflow_gnn.tools.print_training_data:main'
tfgnn_sampled_stats = 'tensorflow_gnn.tools.sampled_stats:main'
tfgnn_validate_graph_schema = 'tensorflow_gnn.tools.validate_graph_schema:main'

[build-system]
requires = ["setuptools", "numpy"]
build-backend = "setuptools.build_meta"

[tool.setuptools.dynamic]
version = {attr = "tensorflow_gnn.version.__version__"}

[tool.setuptools.packages.find]
exclude = ["examples*", "package*", "testdata*", "build*", "bazel*"]

[tool.setuptools.package-data]
tensorflow_gnn = ["*.proto"]