Impulse/pyproject.toml at main · nulib-labs/Impulse · 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
[project]
name = "impulse"
version = "0.1.0"
description = "Superscaling Document Pipelines"
readme = "README.md"


requires-python = ">=3.11,<3.14"
dependencies = [
  "accelerate>=1.12.0",
  "blingfire>=0.1.8",
  "boto3>=1.42.2",
  "bs4>=0.0.2",
  "chandra-ocr>=0.2.0",
  "codecarbon[carbonboard]>=3.2.2",
  "deskew>=1.5.3",
  "dotenv>=0.9.9",
  "fabric>=3.2.2",
  "fireworks>=2.0.8",
  "igraph>=1.0.0",
  "loguru>=0.7.3",
  "lxml>=6.0.2",
  "matplotlib>=3.10.6",
  "mplcursors>=0.7",
  "natsort>=8.4.0",
  "ollama>=0.6.1",
  "opencv-python>=4.12.0.88",
  "pandas>=2.3.3",
  "pdf2image>=1.17.0",
  "pillow>=10.4.0",
  "pip>=25.3",
  "plotly>=6.3.1",
  "pymongo[srv]>=4.15.2",
  "pymupdf>=1.27.1",
  "pypdf2>=3.0.1",
  "python-doctr>=1.0.0",
  "sentence-transformers>=5.1.1",
  "spacy>=3.8.11",
  "surya-ocr>=0.17.1",
  "transformers>=4.57.0",
]

[tool.basedpyright]
typeCheckingMode = "basic"
diagnosticMode = "openFilesOnly"
useLibraryCodeForTypes = false
incude = ["src"]
executionEnvironments = [{ root = "src" }]
venvPath = "."
venv = ".venv"

exclude = [
  "**/node_modules",
  "**/__pycache__",
  "**/.venv",
  "**/.tox",
  "**/dist",
  "**/build",
  "**/.git",
]

[tool.uv.sources]
impulse = { workspace = true }

[dependency-groups]
dev = ["pytest", "pytest-cov", "moto[s3,sqs,ecs]", "mongomock"]

[tool.setuptools.packages.find]
include = ["tasks*"]

[tool.pytest.ini_options]
pythonpath = ["src"]  # tells pytest to add src/ to sys.path