-
Notifications
You must be signed in to change notification settings - Fork 1
/
pyproject.toml
89 lines (79 loc) · 2.76 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
[tool.poetry]
name = "abcdrl"
version = "0.2.0-alpha.6"
description = "Modular Single-file Reinfocement Learning Algorithms Library"
authors = ["Adam Zhao <[email protected]>"]
maintainers = ["Adam Zhao <[email protected]>"]
include = ["abcdrl"]
license= "MIT"
readme = "README.md"
homepage = "https://abcdrl.xyz/"
repository = "https://github.com/sdpkjc/abcdrl"
documentation = "https://docs.abcdrl.xyz/"
keywords = ["reinforcement", "machine", "learning"]
classifiers = [
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Natural Language :: English",
"Natural Language :: Chinese (Simplified)",
"Framework :: Pytest",
"Operating System :: MacOS",
"Operating System :: POSIX :: Linux",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development :: Libraries :: Python Modules"
]
[tool.poetry.urls]
"Bug Tracker" = "https://github.com/sdpkjc/abcdrl/issues"
"Mirror Git Repo" = "https://gitee.com/sdpkjc/abcdrl/"
"Docker Hub" = "https://hub.docker.com/r/sdpkjc/abcdrl/"
[tool.poetry.dependencies]
python = ">=3.8,<3.11"
tensorboard = "^2.10.0"
opencv-python = ">=3.0"
moviepy = ">=1.0.0"
pygame = "2.1.3"
gymnasium = "0.28.1"
wandb = "^0.15.0"
dill = "^0.3.6"
tomli = "^2.0.0"
tyro = "^0.5.0"
wrapt = "1.15.0"
# torch
torch = { version = "^1.12.1", optional = true }
# tensorflow2
tensorflow = { version = "^2.11.0", optional = true }
# atari
shimmy = { version = ">=0.1.0,<1.0", extras = ["atari"], optional = true }
autorom = { version = "~=0.4.2", optional = true }
autorom-accept-rom-license = { version = "~=0.4.2", optional = true }
# mujoco
mujoco = { version = "<=2.3.3", optional = true }
mujoco-py = { version = ">=2.1,<2.2", optional = true }
imageio = { version = ">=2.14.1", optional = true }
free-mujoco-py = { version = "2.1.6", optional = true }
[tool.poetry.extras]
torch = ["torch"]
tf = ["tensorflow"]
atari = ["shimmy", "autorom", "autorom-accept-rom-license"]
mujoco = ["mujoco", "mujoco-py", "imageio", "free-mujoco-py"]
all = ["torch", "tensorflow", "shimmy", "autorom", "autorom-accept-rom-license", "mujoco", "mujoco-py", "imageio", "free-mujoco-py"]
[tool.poetry.group.dev]
optional = true
[tool.poetry.group.dev.dependencies]
pre-commit = "^2.20.0"
pytest = "^7.0.1"
mkdocs-material = "^8.5.11"
mkdocs-glightbox = "^0.3.1"
mkdocs-static-i18n = "^0.53"
mkdocs-git-revision-date-localized-plugin = "^1.1.0"
[build-system]
requires = ["poetry>=1.0.0"]
build-backend = "poetry.masonry.api"
[[tool.poetry.source]]
name = "tsinghua"
url = "https://pypi.tuna.tsinghua.edu.cn/simple/"