-
Notifications
You must be signed in to change notification settings - Fork 1
/
pyproject.toml
136 lines (123 loc) · 3.15 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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
[tool.poetry]
name = "rlhf-trl"
version = "0.1.2"
description = "Reinforcement Learning from Human Feedback with TRL"
authors = ["Victor I. Afolabi <javafolabi@gmail.com>"]
maintainers = ["Victor I. Afolabi <javafolabi@gamil.com>"]
readme = "README.md"
repository = "https://github.com/victor-iyi/rlhf-trl"
keywords = [
"reinforcement-learning",
"human-feedback",
"transformers",
"accelerate",
"pytorch",
]
classifiers = [
"Development Status :: 1 - Planning",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Natural Language :: English",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: Implementation :: CPython",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
packages = [
{include = "rlhf_trl", from = "src"},
]
[tool.poetry.dependencies]
python = "^3.10"
torch = "^2.0.1"
accelerate = "^0.20.3"
transformers = "^4.30.1"
trl = "^0.4.4"
jsonlines = "^3.1.0"
tqdm = "^4.65.0"
datasets = "^2.12.0"
peft = "^0.3.0"
wandb = "^0.15.4"
bitsandbytes = "^0.39.0"
[tool.poetry.group.dev.dependencies]
autopep8 = "^2.0.2"
mypy = "^1.3.0"
pre-commit = "^3.3.2"
flake8 = "^6.0.0"
pydocstyle = {version = "^6.3.0", extras = ["tomli"]}
docformatter = {version = "^1.7.2", extras = ["tomli"]}
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.autopep8]
recursive = true
# in-place = true
diff = true
aggressive = 3
max-line-length = 120
[tool.docformatter]
in-place = true
recursive = true
blank = true
close-quotes-on-newline = true
[tool.pydocstyle]
convention = "google"
add_ignore = [
# Missing docstring in public module.
"D100",
# Missing docstring in public class.
"D101",
# Missing docstring in public package.
"D104",
# Missing docstring in magic method.
"D105",
# Missing docstring in __init__.
"D107",
# No blank lines allowed after function docstring.
"D202",
# 1 blank line required between summary line and description.
"D205",
# First line should end with a period, question mark, or exclamation point.
"D415",
# No blank lines allowed between a section header and its content.
"D412",
# Missing argument descriptions in the docstring.
"D417",
]
[tool.mypy]
files = [
"src/",
]
exclude = [
"tests",
".venv",
"venv",
]
namespace_packages = true
explicit_package_bases = true
strict = true
check_untyped_defs = true
ignore_missing_imports = true
disallow_any_generics = true
disallow_incomplete_defs = true
implicit_reexport = false
no_implicit_optional = true
show_error_codes = true
show_column_numbers = true
allow_untyped_calls = true
# show_error_content = true
warn_unused_configs = true
warn_unused_ignores = true
# warn_redundant_cast = true
warn_return_any = false
warn_unreachable = true
disallow_subclassing_any = false
disallow_untyped_decorators = false
[tool.pyright]
reportOptionalCall = "none"
reportGeneralTypeIssues = "none"
reportOptionalMemberAccess = "none"
reportUnboundVariable = "none"