-
Notifications
You must be signed in to change notification settings - Fork 1
/
pyproject.toml
104 lines (96 loc) · 2.54 KB
/
pyproject.toml
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
[tool.poetry]
name = "pymc_statespace"
version = "0.1.0"
description = ""
authors = ["Nelson Tang <[email protected]>"]
readme = "README.md"
packages = [{include = "pymc_statespace", from="src"}]
[tool.poetry.dependencies]
python = ">=3.10, <4.0"
numpy = "^1.24.3"
pandas = "^2.0.1"
scikit-learn = "^1.2.2"
fastparquet = "^2023.4.0"
numpyro = "^0.13.0"
pymc-experimental = {git = "https://github.com/pymc-devs/pymc-experimental.git"}
ipykernel = "^6.25.2"
statsmodels = "^0.14.0"
orbit-ml = "^1.1.4.2"
graphviz = "^0.20.1"
causalpy = "^0.1.1"
tfcausalimpact = "^0.0.13"
tensorflow = "^2.13.0"
pandas-datareader = "^0.10.0"
statsforecast = "^1.6.0"
sts-jax = {git = "https://github.com/probml/sts-jax"}
ortools = "^9.8.3296"
doubleml = "^0.7.0"
pyro-ppl = "^1.9.0"
tensorflow-probability = "^0.24.0"
tf-keras = "^2.17.0"
[tool.poetry.group.dev.dependencies]
matplotlib = "^3.7.1"
ipykernel = "^6.23.1"
pydantic = "^1.10.7"
pytest = "^7.3.1"
nbformat = "^5.8.0"
ipywidgets = "^8.0.6"
jupytext = "^1.14.5"
jupyter = "^1.0.0"
pre-commit = "^3.3.2"
pymc = "^5.10.3"
[tool.uv]
dev-dependencies = [
"matplotlib<4.0.0,>=3.7.1",
"ipykernel<7.0.0,>=6.23.1",
"pydantic<2.0.0,>=1.10.7",
"pytest<8.0.0,>=7.3.1",
"nbformat<6.0.0,>=5.8.0",
"ipywidgets<9.0.0,>=8.0.6",
"jupytext<2.0.0,>=1.14.5",
"jupyter<2.0.0,>=1.0.0",
"pre-commit<4.0.0,>=3.3.2",
"pymc<6.0.0,>=5.10.3",
]
[tool.pdm.build]
includes = [
"src/pymc_statespace",
]
[build-system]
requires = ["pdm-backend"]
build-backend = "pdm.backend"
[project]
authors = [
{name = "Nelson Tang", email = "[email protected]"},
]
requires-python = "<3.11,>=3.10"
dependencies = [
"numpy<2.0.0,>=1.24.3",
"pandas<3.0.0,>=2.0.1",
"scikit-learn<2.0.0,>=1.2.2",
"fastparquet<2024.0.0,>=2023.4.0",
"numpyro<1.0.0,>=0.13.0",
"pymc-experimental @ git+https://github.com/pymc-devs/pymc-experimental.git",
"ipykernel<7.0.0,>=6.25.2",
"statsmodels<1.0.0,>=0.14.0",
"orbit-ml<2.0.0.0,>=1.1.4.2",
"graphviz<1.0.0,>=0.20.1",
"causalpy<1.0.0,>=0.1.1",
"tfcausalimpact<1.0.0,>=0.0.13",
"tensorflow<3.0.0,>=2.13.0",
"pandas-datareader<1.0.0,>=0.10.0",
"statsforecast<2.0.0,>=1.6.0",
"sts-jax @ git+https://github.com/probml/sts-jax",
"ortools<10.0.0,>=9.8.3296",
"doubleml<1.0.0,>=0.7.0",
"pyro-ppl<2.0.0,>=1.9.0",
"tensorflow-probability<1.0.0,>=0.24.0",
"tf-keras<3.0.0,>=2.17.0",
"jax>=0.4.35",
"polars>=1.19.0",
"pymc>=5.18.1",
]
name = "pymc_statespace"
version = "0.1.0"
description = ""
readme = "README.md"