gnn/pyproject.toml at c2a6fcfe62c1653dd1e7d9e3d8cbe0b9420d5ed3 · 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
[project]
name = "tensorflow_gnn"
description = "A library for building scalable graph neural networks in TensorFlow."
readme = "README.md"
keywords = ["tensorflow", "gnn", "graph"]
dynamic = ["version"]
license = "Apache-2.0"
requires-python = ">=3.9, <4"
dependencies = [
  "google-vizier>=0.0.13, !=0.1.23, !=0.1.24",  # b/394062744
  "ml-collections>=1.1.0",
  "networkx>=3.6.1",
  "pyarrow>=18.1.0",
  "tensorflow>=2.12.0, <3; platform_machine != 'arm64' or platform_system != 'Darwin'",
  "tensorflow-macos>=2.12.0, <3; platform_machine == 'arm64' and platform_system == 'Darwin'",
  "apache-beam>=2.54",
  "crcmod>=1.7, <2",
]
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.9",
  "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"

[dependency-groups]
test-base = [
  "mock",
]
test-pre-tf216 = [
  "ai-edge-litert",
  {include-group = "test-base"},
]
test-tf216plus = [
  "tf-keras",
  {include-group = "test-pre-tf216"},
]
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"]