forked from ai4co/rl4co
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
143 lines (128 loc) · 3.28 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
137
138
139
140
141
142
143
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"
[tool.setuptools.packages.find]
include = ['rl4co*']
# Automatically determine version from package
[tool.setuptools.dynamic]
version = {attr = "rl4co.__version__"}
[project]
name = "rl4co"
readme = "README.md"
description = "RL4CO: a Unified Reinforcement Learning for Combinatorial Optimization Library"
authors = [
{name = "Federico Berto", email = "[email protected]"},
{name = "Chuanbo Hua", email="[email protected]"},
{name = "Junyoung Park", email="[email protected]"},
]
dynamic = ["version"]
license = {file = "LICENSE"}
requires-python = ">=3.8"
classifiers = [
"Programming Language :: Python",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"License :: OSI Approved :: Apache Software License",
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"Operating System :: OS Independent",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
"torchrl>=0.2.0",
"tensordict>=0.2.0",
"lightning>=2.1.0",
"hydra-core",
"hydra-colorlog",
"omegaconf",
"pyrootutils",
"rich",
"einops",
"wandb",
"matplotlib",
"scipy",
"robust-downloader"
]
[project.optional-dependencies]
graph = ["torch_geometric"]
testing = ["pytest", "pytest-cov"]
dev = ["black", "ruff", "pre-commit>=3.3.3"]
[project.urls]
"Homepage" = "https://github.com/ai4co/rl4co"
"Bug Tracker" = "https://github.com/ai4co/rl4co/issues"
# Ruff + Black + isort combo for code formatting
# Adapted from https://github.com/AntonOsika/gpt-engineer (kudos!)
# https://beta.ruff.rs/docs/configuration/#using-pyprojecttoml
[tool.ruff]
select = ["F", "E", "W", "I001"]
line-length = 90
show-fixes = false
target-version = "py310"
task-tags = ["TODO", "FIXME"]
exclude = [
".bzr",
".direnv",
".eggs",
".git",
".ruff_cache",
".svn",
".tox",
".venv",
"__pypackages__",
"_build",
"buck-out",
"build",
"dist",
"node_modules",
"venv",
]
ignore = ["E501"] # never enforce `E501` (line length violations), done in Black
# Ignore `E402` (import violations) and "F401" (unused imports) in all `__init__.py` files
[tool.ruff.per-file-ignores]
"__init__.py" = ["E402", "F401"]
[tool.ruff.isort]
known-first-party = []
known-third-party = []
section-order = [
"future",
"standard-library",
"third-party",
"first-party",
"local-folder",
]
combine-as-imports = true
split-on-trailing-comma = false
lines-between-types = 1
# https://black.readthedocs.io/en/stable/usage_and_configuration/the_basics.html
[tool.black]
line-length = 90
target-version = ["py311"]
include = '\.pyi?$'
exclude = '''
(
/(
\.direnv
| \.eggs
| \.git
| \.tox
| \.venv
| _build
| build
| dist
| venv
)/
)
'''
[tool.coverage]
include = ["rl4co.*"]
[tool.coverage.report]
show_missing = true
exclude_lines = [
# Lines to exclude from coverage report (e.g., comments, debug statements)
"pragma: no cover",
"if __name__ == .__main__.:",
]