xuance/pyproject.toml at master · agi-brain/xuance · 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
[build-system]
requires = ["setuptools>=61", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "xuance"
version = "1.4.1"
description = "XuanCe: A Comprehensive and Unified Deep Reinforcement Learning Library."
readme = "README.md"
requires-python = ">=3.6"
license = { text = "MIT" }

authors = [
  { name = "Wenzhang Liu", email = "liu_wzh@foxmail.com" }
]

keywords = [
  "deep reinforcement learning",
  "open source library",
  "PyTorch",
  "TensorFlow2",
  "MindSpore"
]

classifiers = [
  "Development Status :: 4 - Beta",
  "Intended Audience :: Developers",
  "Topic :: Software Development :: Build Tools",
  "License :: OSI Approved :: MIT License",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.6",
  "Programming Language :: Python :: 3.7",
  "Programming Language :: Python :: 3.8",
  "Programming Language :: Python :: 3.9",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
]

dependencies = [
  "numpy",
  "scipy",
  "PyYAML",
  "gymnasium",
  "pygame",
  "tqdm",
  "pyglet==1.5.15",
  "pettingzoo",
  "tensorboard",
  "wandb",
  "moviepy==1.0.3",
  "imageio",
  "opencv-python",
]

[project.optional-dependencies]
torch = ["torch", "torchvision"]
tensorflow = ["tensorflow==2.19.0", "tf_keras==2.19.0"]
mindspore = ["mindspore==2.6.0", "protobuf==3.20.0"]

all = [
  "torch",
  "torchvision",
  "tensorflow==2.19.0",
  "tf_keras==2.19.0",
  "mindspore==2.6.0",
  "protobuf==3.20.0"
]

tune = [
  "optuna>=4.1.0",
  "optuna-dashboard>=0.17.0",
  "plotly>=5.24.1"
]

mujoco = ["gymnasium[mujoco]"]
atari = [
  "gymnasium[accept-rom-license]",
  "gymnasium[atari]",
  "ale-py==0.10.1"
]
box2d = [
  "swig==4.3.0",
  "gymnasium[box2d]"
]
minigrid = ["minigrid==3.0.0"]
metadrive = ["metadrive"]
rware = ["rware"]
einops = ["einops==0.8.1"]

[tool.setuptools.packages.find]
include = ["xuance", "xuance.*"]

[tool.setuptools.package-data]
xuance = [
  "configs/*.yaml",
  "configs/*/*.yaml",
  "configs/*/*/*.yaml",
  "environment/magent2/libmagent.so",
  "environment/magent2/magent.dll",
  "environment/magent2/libmagent.dylib"
]