/
pyproject.toml
61 lines (54 loc) · 1.36 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
[tool.poetry]
name = "casus"
version = "0.0.1"
description = ""
authors = ["Rishikanth Chandrasekaran <r3chandr@ucsd.edu>"]
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.10"
jax = "*"
jaxlib = "*"
# jaxlib = {version = "^0.4.23+cuda11.cudnn86", source = "jax"} #uncomment for cuda support
equinox = "*"
numpy = "*"
loguru = "*"
einops = "^0.7.0"
quax = "^0.0.3"
[tool.poetry.group.dev.dependencies]
black = "*"
pyright = "*"
ruff = "*"
pre-commit = "*"
pytest = "*"
fire = "*"
marimo = "^0.2.3"
holoviews = "^1.18.2"
bokeh = "^3.3.4"
torch = "^2.2.0"
torchvision = "^0.17.0"
alive-progress = "^3.1.5"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.pyright]
venvPath = "/Users/rishi/miniconda3/envs"
venv = "casus"
useLibraryCodeForTypes = true
typeCheckingMode = "basic"
reportUnusedImport = "warning"
reportUnusedVariable = "warning"
reportDuplicateImport = "error"
reportPrivateUsage = "warning"
reportDeprecated = "warning"
reportInconsistentConstructor = "error"
reportMissingSuperCall = "error"
reportUninitializedInstanceVariable = "error"
reportUnnecessaryCast = "warning"
reportUnnecessaryComparison = "warning"
reportImplicitStringConcatenation = "warning"
exclude = ["**/notebooks/"]
[tool.pytest.ini_options]
testpaths = ["tests"]
markers = ["hv", "emb", "ml"]
[tool.poetry.scripts]
t-mnist = "tests.test_ml:main"