-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpyproject.toml
38 lines (34 loc) · 969 Bytes
/
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
[tool.poetry]
name = "sequifier"
version = "0.4.1.0"
authors = ["Leon Luithlen <[email protected]>"]
description = "Train a transformer model with the command line"
keywords = ["transformer", "sequence classification", "machine learning", "sequence", "sequence modelling", "nlp", "language", "language modelling", "torch", "pytorch"]
readme = "README.md"
license = "BSD 3-Clause"
homepage = "https://github.com/0xideas/sequifier"
repository = "https://github.com/0xideas/sequifier"
include = [
"LICENSE",
]
[tool.poetry.dependencies]
python = "^3.10"
pandas = "^2.0"
numpy = "^1.23"
torch = "^2.0"
torch_optimizer = "^0.3.0"
onnx = "^1.15.0"
onnxruntime = "^1.17"
pyyaml = "^6.0"
pydantic = "1.*"
poetry = "^1.3"
pytest = "^7.2"
csvkit = "^1.0"
pyarrow = "^15.0"
fastparquet = "^2024.2.0"
beartype = "^0.18.5"
[tool.poetry.scripts]
sequifier = 'sequifier.sequifier:main'
[build-system]
requires = ["poetry>=0.12"]
build-backend = "poetry.masonry.api"