meridian/pyproject.toml at main · google/meridian · 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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
[project]
# Project metadata. Available keys are documented at:
# https://packaging.python.org/en/latest/specifications/declaring-project-metadata
name = "google-meridian"
description = """\
  Google's open source mixed marketing model library, helps you understand \
  your return on investment and direct your ad spend with confidence. \
  """
readme = "README.md"
requires-python = ">=3.10"
license-files = ["LICENSE"]
authors = [
  {name = "The Meridian Authors", email="no-reply@google.com"},
]
classifiers = [  # List of https://pypi.org/classifiers/
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3 :: Only",
    "Topic :: Other/Nonlisted Topic",
    "Topic :: Scientific/Engineering :: Information Analysis",
]
keywords = [
  "mmm",
]
# pip dependencies of the project
# Installed locally with `pip install -e .`
dependencies = [
  "arviz < 0.20.0",
  "altair >= 5",
  "bidict",
  "immutabledict",
  "joblib",
  "natsort >= 7.1.1, < 8",
  "numpy >= 2.0.2, < 2.4.0",
  "pandas >= 2.2.2, < 3",
  "scipy >= 1.13.1, < 2",
  "statsmodels >= 0.14.5",
  "tensorflow >= 2.19.0, < 2.21",
  "tfp-nightly == 0.26.0.dev20260130",
  "tf-keras >= 2.18, < 2.21",
  "xarray",
]
# `version` is automatically set by flit to use `meridian.__version__`
dynamic = ["version"]

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

[project.optional-dependencies]
# Development deps (unittest, linting, formating,...)
# Installed through `pip install -e .[dev]`
dev = [
  "pytest>=8.0.0",
  "pytest-xdist",
  "pylint>=2.6.0",
  "pyink",
]
# Colab deps
# Installed through `pip install -e .[colab]`
colab = [
  "psutil",
  "python-calamine",  # For parsing the included simulated .xlsx files.
]
# GPU deps
# Installed through `pip install -e .[and-cuda]`
and-cuda = [
  "tensorflow[and-cuda] >= 2.19.0, < 2.21",
]
# MLflow deps
# Installed through `pip install -e .[mlflow]`
mlflow = ["mlflow"]
# JAX backend dependencies.
# Installed through `pip install -e .[jax]`
jax = [
    "jax==0.7.2",
    "jaxlib==0.7.2",
    "tfp-nightly[substrates-jax]==0.26.0.dev20260130",
]
# MMM proto schema.
schema = [
  "mmm-proto-schema >= 1.1.3",
  "semver",
]

# Meridian Scenario Planner.
scenarioplanner = [
  "google-api-python-client",
  "google-auth",
  "mmm-proto-schema >= 1.1.3",
  "semver",
]

[tool.pyink]
# Formatting configuration to follow Google style-guide
line-length = 80
preview = true
pyink-indentation = 2
pyink-use-majority-quotes = true

[build-system]
# Build system specify which backend is used to build/install the project (flit,
# poetry, setuptools,...). All backends are supported by `pip install`
requires = [
  "setuptools >= 61.0.0",
  "libsass",
]
build-backend = "setuptools.build_meta"

[tool.setuptools]
include-package-data = true
py-modules = ["schema"]  # TODO: Remove this shim.

[tool.setuptools.packages.find]
include = ["meridian*", "scenarioplanner*"]
exclude = ["*test"]

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