atdata/pyproject.toml at main · forecast-bio/atdata · 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
[project]
name = "atdata"
version = "0.7.0b1"
description = "A loose federation of distributed, typed datasets"
readme = "README.md"
authors = [
    { name = "Maxine Levesque", email = "hello@maxine.science" },
    { name = "Maxine @ Forecast Bio", email = "maxine@forecast.bio" }
]
requires-python = ">=3.12"
dependencies = [
    "atproto>=0.0.65",
    "boto3>=1.41.5",
    "fastparquet>=2024.11.0",
    "libipld>=3.3.2",
    "msgpack>=1.1.2",
    "numpy>=2.3.4",
    "ormsgpack>=1.11.0",
    "pandas>=2.3.3",
    "pydantic>=2.12.5",
    "python-dotenv>=1.2.1",
    "redis-om>=0.3.5",
    "requests>=2.32.5",
    "s3fs>=2025.12.0",
    "schemamodels>=0.9.1",
    "tqdm>=4.67.1",
    "typer>=0.21.1",
    "webdataset>=1.0.2",
]

[project.optional-dependencies]
atmosphere = [
    "atproto>=0.0.55",
]
postgres = [
    "psycopg[binary]>=3.1",
]
sparse = [
    "scipy>=1.10",
]
arrow = [
    "pyarrow>=14.0",
]
safetensors = [
    "safetensors>=0.4",
]
all-formats = [
    "scipy>=1.10",
    "pyarrow>=14.0",
    "safetensors>=0.4",
]

[project.scripts]
atdata = "atdata:main"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.sdist]
exclude = [
    ".chainlink/",
    ".claude/",
]

[tool.hatch.build.targets.wheel]
packages = ["src/atdata"]

[tool.pytest.ini_options]
addopts = "--cov=atdata --cov-report=html -m 'not integration'"
markers = [
    "network: marks tests as requiring network access (deselect with '-m \"not network\"')",
    "slow: marks tests as slow running",
    "integration: live integration tests against remote service sandboxes (deselect with '-m \"not integration\"')",
    "benchmark: performance benchmarks",
    "bench_serial: serialization benchmarks (μs scale)",
    "bench_index: index provider benchmarks (μs–ms scale)",
    "bench_io: shard read/write benchmarks (ms scale)",
    "bench_query: query system benchmarks (ms scale)",
    "bench_s3: S3 storage benchmarks via moto (ms+ scale)",
]

[dependency-groups]
dev = [
    "jupyter>=1.1.1",
    "matplotlib>=3.10",
    "moto[s3]>=5.0.29",
    "pytest>=8.4.2",
    "pytest-cov>=7.0.0",
    "quartodoc>=0.11.1",
    "pytest-benchmark>=5.1.0",
    "ruff>=0.14.13",
]