dpsynth/pyproject.toml at main · google/dpsynth · 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
93
94
95
96
[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"

[project]
name = "dpsynth"
description = "Differentially Private Synthetic Tabular Data Generation"
readme = "README.md"
requires-python = ">=3.12,<3.14"
license = {file = "LICENSE"}
authors = [{name = "dpsynth authors", email="mckennar@google.com"}]
classifiers = [
    "License :: OSI Approved :: Apache Software License",
    "Intended Audience :: Science/Research",
]
keywords = []
dependencies = [
    "absl-py",
    "attrs",
    "cattrs",
    "etils[epath]",
    "fancyflags",
    "numpy",
    "pandas",
    "pydantic>=2.0",
    "pyyaml",
    "dp_accounting @ git+https://github.com/google/differential-privacy.git#subdirectory=python/dp_accounting",
    "jax[cpu]==0.11.0",
    "more-itertools",
    "networkx",
    "scipy",
    "tqdm",
    "mbi @ git+https://github.com/ryan112358/mbi.git",
]
dynamic = ["version"]

[project.urls]
homepage = "https://github.com/google/dpsynth"
repository = "https://github.com/google/dpsynth"
changelog = "https://github.com/google/dpsynth/blob/main/CHANGELOG.md"

[project.optional-dependencies]
pipeline = [
    "apache-beam>=2.75.0",
    "google-crc32c",
    "pipeline-dp",
    "tensorflow",
]
all = [
    "dpsynth[pipeline,text]",
]
dev = [
    "chex",
    "pytest",
    "pytest-xdist",
    "pyrefly",
]
text = [
    "flax",
    "gemma",
    "google-genai",
    "jax-privacy @ git+https://github.com/google-deepmind/jax_privacy.git",
    "optax",
]
docs = [
    "sphinx",
    "sphinx-rtd-theme",
    "myst-parser",
]
examples = [
    "dpsynth[text]",
    "datasets",
    "kagglehub",
    "scikit-learn",
    "sdmetrics",
]

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

[tool.setuptools.packages.find]
namespaces = false

[tool.pytest.ini_options]
addopts = "--import-mode=importlib"
pythonpath = ["."]
filterwarnings = [
    "ignore:.*JAX is running in float32 mode.*:UserWarning",
    "ignore:.*JAX persistent compilation cache is enabled.*:UserWarning",
]

[tool.pyrefly]
project-includes = [
  "**/*.py",
]
preset = "basic"