VulnTrain/pyproject.toml at main · vulnerability-lookup/VulnTrain · 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
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
[build-system]
requires = ["poetry-core>=2.0.0,<3.0.0"]
build-backend = "poetry.core.masonry.api"


[project]
name = "VulnTrain"
version = "3.1.0"
description = "Generate datasets amd models based on vulnerabilities data from Vulnerability-Lookup."
authors = [
    {name = "Cédric Bonhomme", email = "cedric.bonhomme@circl.lu"},
    {name = "Léa Ulusan", email = "lea.ulusan@edu.ece.fr"}
]
license = "GPL-3.0-or-later"
readme = "README.md"

dynamic = ["classifiers"]

requires-python = ">=3.11,<3.15"
dependencies = [
    "nltk (>=3.9.2)",
    "pandas (>=3.0.0,<4.0.0)",
    "datasets (>=4.5.0,<5.0.0)",
    "valkey (>=6.1.1)",
    "torch (>=2.10.0,<3.0.0)",
    "transformers[torch] (>=5.0.0,<6.0.0)",
    "evaluate (>=0.4.6,<0.5.0)",
    "codecarbon (>=3.2.1,<4.0.0)",
    "scikit-learn (>=1.8.0,<2.0.0)",
    "markdown-it-py (>=3.0.0,<4.0.0)",
    "cvss (>=3.6,<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.generation_description:main"
vulntrain-train-severity-classification = "vulntrain.trainers.classify_severity:main"
vulntrain-train-severity-cnvd-classification = "vulntrain.trainers.classify_severity_cnvd:main"
vulntrain-train-cwe-classification = "vulntrain.trainers.cwe_guesser_patches: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.poetry.group.docs.dependencies]
sphinx = "^8.2.3"
sphinx-book-theme = "^1.1.4"
myst-parser = "^4.0.1"


[tool.poetry.group.docs]
optional = true


[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"