-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpyproject.toml
90 lines (75 loc) · 2.44 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
[build-system]
requires = ["poetry-core>=2.0.0,<3.0.0"]
build-backend = "poetry.core.masonry.api"
[project]
name = "VulnTrain"
version = "1.2.0"
description = "Generate datasets amd models based on vulnerabilities data from Vulnerability-Lookup."
authors = [
{name = "Cédric Bonhomme",email = "[email protected]"}
]
license = "GPL-3.0-or-later"
readme = "README.md"
dynamic = ["classifiers"]
requires-python = ">=3.10,<4.0"
dependencies = [
"nltk (>=3.9.1)",
"pandas (>=2.2.3)",
"datasets (>=3.3.1)",
"valkey (>=6.1.0)",
"torch (>=2.6.0)",
"transformers[torch] (>=4.49.0,<5.0.0)",
"evaluate (>=0.4.3,<0.5.0)",
"codecarbon (>=2.8.3,<3.0.0)",
"scikit-learn (>=1.6.1,<2.0.0)",
"markdown-it-py (>=3.0.0,<4.0.0)",
"cvss (>=3.4,<4.0)"
]
[project.urls]
Homepage = "https://github.com/vulnerability-lookup/VulnTrain"
Repository = "https://github.com/vulnerability-lookup/VulnTrain"
Changelog = "https://github.com/vulnerability-lookup/VulnTrain/blob/main/CHANGELOG.md"
[project.scripts]
# Commands for datasets generation:
vulntrain-dataset-generation = "vulntrain.datasets.create_dataset:main"
# Commands for trainings:
vulntrain-train-description-generation = "vulntrain.trainers.summarize:main"
vulntrain-train-severity-classification = "vulntrain.trainers.classify:main"
# Commands for validations:
vulntrain-validate-text-generation = "vulntrain.validators.summarize:main"
vulntrain-validate-severity-classification = "vulntrain.validators.severity:main"
[tool.poetry]
requires-poetry = ">=2.0"
classifiers = [
"Development Status :: 5 - Production/Stable",
"Environment :: Console",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"Topic :: Security",
"Operating System :: POSIX :: Linux",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Programming Language :: Python :: 3.13",
"License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)"
]
include = [
"README.md",
"COPYING",
]
[tool.mypy]
python_version = "3.13"
check_untyped_defs = true
ignore_errors = false
ignore_missing_imports = true
strict_optional = true
no_implicit_optional = true
warn_unused_ignores = true
warn_redundant_casts = true
warn_unused_configs = true
warn_unreachable = true
show_error_context = true
pretty = true
exclude = "build|dist|docs"
[tool.isort]
profile = "black"