-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
56 lines (49 loc) · 1.26 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
[tool.poetry]
name = "tunnrl_tbm_maintenance"
version = "0.1.0"
description = "Functionality to train, optimize and execute reinforcement learning models controlling the maintenance of cutter discs for a Tunnel Boring Machine (TBM). A simulated TBM environment is defined."
authors = ["geograz <->", "tfha <[email protected]>"]
license = "MIT"
[tool.poetry.dependencies]
python = ">=3.10.5,<3.12"
stable-baselines3 = "^1.6.0"
numpy = "^1.23.1"
optuna = "^2.10.1"
joblib = "^1.1.0"
scikit-learn = "^1.1.1"
importlib-resources = "^5.9.0"
pysqlite3 = "^0.4.7"
mlflow = "^1.28.0"
tensorboard = "^2.10.0"
umap-learn = "^0.5.3"
[tool.poetry.dev-dependencies]
mypy = "^0.971"
rich = "^12.5.1"
setuptools = "<60"
isort = "^5.10.1"
flake8 = "^5.0.3"
flake8-annotations = "^2.9.1"
ipdb = "^0.13.9"
black = "^22.6.0"
pyupgrade = "^2.37.3"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"
[tool.ipdb]
context = 8
[tool.isort]
profile = "black"
known_first_party = "src"
force_single_line = false
lines_after_imports = 2
[tool.black]
line-length = 79
[tool.mypy]
strict = false
strict_optional = false
warn_unreachable = true
pretty = true
show_column_numbers = true
show_error_codes = true
show_error_context = true
ignore_missing_imports = true