-
Notifications
You must be signed in to change notification settings - Fork 1
/
pyproject.toml
103 lines (84 loc) · 2.75 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
[tool.poetry]
name = "emma-policy"
version = "1.48.0"
description = ""
authors = ["Amit Parekh <amit.parekh@hw.ac.uk>"]
readme = "README.md"
repository = "https://github.com/emma-heriot-watt/policy"
packages = [{ include = "emma_policy", from = "src" }]
[tool.poe.tasks]
[tool.poe.tasks.format]
help = "Format using the pre-commit hooks"
cmd = "pre-commit run --all-files"
[tool.poe.tasks.typecheck]
help = "Check types with mypy"
cmd = "mypy ."
[tool.poe.tasks.lint]
help = "Lint with flake8"
cmd = "flake8 ."
[tool.poe.tasks.test]
help = "Run the fast Python tests"
cmd = "pytest --cov=src -m 'not slow'"
[tool.poe.tasks.test-everything]
help = "Run all the tests and get the coverage"
cmd = "pytest -v --durations=40 --color=yes --junitxml=pytest.xml --cov=src -m 'not slow' -n 2"
[tool.poe.tasks.update-torch-cuda]
## See https://github.com/python-poetry/poetry/issues/2543
help = "Update torch to use the latest CUDA version"
shell = "python scripts/update_torch_cuda.py"
[[tool.poe.tasks.download-training-data]]
help = "Download mscoco_resplit_train.json"
shell = "wget -O storage/constants/mscoco_resplit_train.json https://www.dropbox.com/sh/m26a281y97ok6wm/AAA-97WOmyvcWLaaf2jFRrT8a/mscoco_resplit_train.json\\?dl\\=0"
interpreter = "bash"
[tool.poetry.dependencies]
python = ">=3.9,<3.10"
torch = "1.11.0"
torchvision = "0.12.0"
pytorch-lightning = "1.5.10"
transformers = "4.18.0"
torchmetrics = "0.7.3"
wandb = ">=0.12.17"
hydra-core = ">=1.1.2"
hydra-colorlog = ">=1.1.0"
overrides = ">=6.1.0"
numpy = ">=1.22.2,<1.24"
scipy = ">=1.8.0"
spacy = "3.7.2"
pydantic = { extras = ["dotenv"], version = ">=1.10.2,<2" }
emma-datasets = { git = "https://github.com/emma-heriot-watt/datasets.git" }
emma-common = { git = "https://github.com/emma-heriot-watt/common.git" }
en-core-web-lg = { url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_lg-3.7.1/en_core_web_lg-3.7.1-py3-none-any.whl" }
[tool.poetry.group.dev.dependencies]
wemake-python-styleguide = ">=0.16.0"
pytest = ">=7.1.1"
ipykernel = ">=6.13.0"
pre-commit = ">=2.18.1"
poethepoet = ">=0.13.1"
pytest-cov = ">=3.0.0"
jupyterlab = ">=3.3.4"
pudb = ">=2022.1"
isort = ">=5.10.1"
pytest-cases = ">=3.6.11"
black = { version = ">=22.3.0", extras = ["jupyter"] }
pytest-order = ">=1.0.1"
mypy = ">=0.950"
types-requests = ">=2.27.19"
pytest-mock = ">=3.7.0"
types-mock = ">=4.0.13"
flake8-pytest-style = ">=1.6.0"
pytest-xdist = ">=2.5.0"
filelock = ">=3.6.0"
requests-mock = ">=1.9.3"
[tool.poetry.group.demo.dependencies]
gradio = ">=3.7"
[tool.black]
line-length = 99
[tool.isort]
profile = "black"
combine_as_imports = true
lines_after_imports = 2
line_length = 99
known_first_party = ["emma_policy"]
[build-system]
requires = ["poetry-core>=1.0.0", "setuptools"]
build-backend = "poetry.core.masonry.api"