unienv 0.0.1b1__py3-none-any.whl → 0.0.1b2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- unienv-0.0.1b2.dist-info/METADATA +73 -0
- {unienv-0.0.1b1.dist-info → unienv-0.0.1b2.dist-info}/RECORD +5 -5
- unienv-0.0.1b1.dist-info/METADATA +0 -20
- {unienv-0.0.1b1.dist-info → unienv-0.0.1b2.dist-info}/WHEEL +0 -0
- {unienv-0.0.1b1.dist-info → unienv-0.0.1b2.dist-info}/licenses/LICENSE +0 -0
- {unienv-0.0.1b1.dist-info → unienv-0.0.1b2.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1,73 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: unienv
|
|
3
|
+
Version: 0.0.1b2
|
|
4
|
+
Summary: Unified robot environment framework supporting multiple tensor and simulation backends
|
|
5
|
+
License-Expression: MIT
|
|
6
|
+
Project-URL: Homepage, https://www.quantumcookie.xyz/project/unienv/
|
|
7
|
+
Project-URL: Documentation, https://readthedocs.org
|
|
8
|
+
Project-URL: Repository, https://github.com/UniEnvOrg/UniEnvPy
|
|
9
|
+
Project-URL: Issues, https://github.com/UniEnvOrg/UniEnvPy/issues
|
|
10
|
+
Project-URL: Changelog, https://github.com/UniEnvOrg/UniEnvPy/blob/main/CHANGELOG.md
|
|
11
|
+
Requires-Python: >=3.10
|
|
12
|
+
Description-Content-Type: text/markdown
|
|
13
|
+
License-File: LICENSE
|
|
14
|
+
Requires-Dist: numpy
|
|
15
|
+
Requires-Dist: xbarray>=0.0.1a8
|
|
16
|
+
Requires-Dist: pillow
|
|
17
|
+
Requires-Dist: h5py
|
|
18
|
+
Provides-Extra: dev
|
|
19
|
+
Requires-Dist: pytest; extra == "dev"
|
|
20
|
+
Provides-Extra: gymnasium
|
|
21
|
+
Requires-Dist: gymnasium>=0.29.0; extra == "gymnasium"
|
|
22
|
+
Provides-Extra: video
|
|
23
|
+
Requires-Dist: moviepy>=2.1; extra == "video"
|
|
24
|
+
Provides-Extra: mjx
|
|
25
|
+
Requires-Dist: playground; extra == "mjx"
|
|
26
|
+
Provides-Extra: maniskill
|
|
27
|
+
Requires-Dist: mani_skill>=3.0.0b12; extra == "maniskill"
|
|
28
|
+
Dynamic: license-file
|
|
29
|
+
|
|
30
|
+
# UniEnvPy
|
|
31
|
+
|
|
32
|
+
TLDR: Gymnasium Library replacement with support for multiple tensor backends
|
|
33
|
+
|
|
34
|
+
Provides an universal interface for single / parallel state-based or function-based environments. Also contains a set of utilities (such as replay buffers, wrappers, etc.) to facilitate the training of reinforcement learning agents.
|
|
35
|
+
|
|
36
|
+
## Cross-backend Support
|
|
37
|
+
|
|
38
|
+
UniEnvPy supports multiple tensor backends with zero-copy translation layers through the DLPack protocol, and allows you to use the same abstract compute backend interface to write custom data transformation layers, environment wrappers and other utilities. This is powered by the [xbarray](https://github.com/realquantumcookie/xbarray) package, which builts on top of the Array API Standard, and supports the following backends:
|
|
39
|
+
|
|
40
|
+
- numpy
|
|
41
|
+
- pytorch
|
|
42
|
+
- jax
|
|
43
|
+
|
|
44
|
+
We also support diverse simulation environments and real robots, built on top of the abstract environment / world interface. This allows you to reuse code across different sim and real robots.
|
|
45
|
+
|
|
46
|
+
Current supported simulation environments:
|
|
47
|
+
- Any Environment defined in Gymnasium interface
|
|
48
|
+
- <s>Mujoco</s> (New code will be added in the future, but I'm currently working on refractoring World based environments)
|
|
49
|
+
- MJX based on [Mujoco-Playground](https://github.com/google-deepmind/mujoco_playground)
|
|
50
|
+
- [ManiSkill 3](https://github.com/haosulab/ManiSkill/)
|
|
51
|
+
|
|
52
|
+
Current supported real robots:
|
|
53
|
+
- Franka Research 3 + RobotiQ Gripper in Droid Setup
|
|
54
|
+
- OyMotion OHand
|
|
55
|
+
|
|
56
|
+
## Installation
|
|
57
|
+
|
|
58
|
+
Install the package with pip
|
|
59
|
+
|
|
60
|
+
```bash
|
|
61
|
+
pip install unienv
|
|
62
|
+
```
|
|
63
|
+
|
|
64
|
+
You can install optional dependencies such as `gymnasium`, `mjx`, `maniskill`, `video` by running
|
|
65
|
+
|
|
66
|
+
```bash
|
|
67
|
+
pip install unienv[gymnasium,mjx,maniskill,video]
|
|
68
|
+
```
|
|
69
|
+
|
|
70
|
+
## Acknowledgements
|
|
71
|
+
|
|
72
|
+
The idea of this project is inspired by [Gymnasium](https://github.com/Farama-Foundation/Gymnasium) and its predecessor [OpenAI Gym](https://github.com/openai/gym).
|
|
73
|
+
This library is impossible without the great work of DataAPIs Consortium and their work on the [Array API Standard](https://data-apis.org/array-api/latest/). The zero-copy translation layers are powered by the [DLPack](https://github.com/dmlc/dlpack) project.
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
unienv-0.0.
|
|
1
|
+
unienv-0.0.1b2.dist-info/licenses/LICENSE,sha256=VHeh-ceoc7FrtFmwszuGP2ZgOQGJNXIIWggxpk3B54E,1130
|
|
2
2
|
unienv_data/__init__.py,sha256=zFxbe7aM5JvYXIK0FGnOPwWQJMN-8l_l8prB85CkcA8,95
|
|
3
3
|
unienv_data/base/__init__.py,sha256=4RP84bYe8lpDFF3Ow1Ch4P91HGgE0Y_nmfEj1Ah5nxE,232
|
|
4
4
|
unienv_data/base/common.py,sha256=SBYd2i-ZZq7ws_vphn8yx-q07jAbmMqUDvq4rdVlADE,10179
|
|
@@ -79,7 +79,7 @@ unienv_maniskill/__init__.py,sha256=GJ6Fe5XosHqVxXSQ_cZT6cd9GWmfb2_NIYefCFD7rpU,
|
|
|
79
79
|
unienv_maniskill/wrapper/maniskill_compat.py,sha256=yp9kX3Sn5Riszai47VasQcbGD6MFs67ITl0NuFqxp7Y,7435
|
|
80
80
|
unienv_mjxplayground/__init__.py,sha256=EaiB9FV7um1oZhhUNPx-hf4j3ruQ51Q1OSiS7Njls1M,59
|
|
81
81
|
unienv_mjxplayground/wrapper/playground_compat.py,sha256=xbkJ_7kO1VQjYsrtxT9Er6NNls4H5o5RZAe0ttgWX5w,7945
|
|
82
|
-
unienv-0.0.
|
|
83
|
-
unienv-0.0.
|
|
84
|
-
unienv-0.0.
|
|
85
|
-
unienv-0.0.
|
|
82
|
+
unienv-0.0.1b2.dist-info/METADATA,sha256=u-sC2zwEoXuF2C23dh2aZZPju2WgpA6weQmyPMM6aYs,3216
|
|
83
|
+
unienv-0.0.1b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
84
|
+
unienv-0.0.1b2.dist-info/top_level.txt,sha256=lL1n-OMi2oZ5e4kuFcvbelQ-9DsdEgr7w-dQ-faiCD8,67
|
|
85
|
+
unienv-0.0.1b2.dist-info/RECORD,,
|
|
@@ -1,20 +0,0 @@
|
|
|
1
|
-
Metadata-Version: 2.4
|
|
2
|
-
Name: unienv
|
|
3
|
-
Version: 0.0.1b1
|
|
4
|
-
Requires-Python: >=3.10
|
|
5
|
-
License-File: LICENSE
|
|
6
|
-
Requires-Dist: numpy
|
|
7
|
-
Requires-Dist: xbarray>=0.0.1a8
|
|
8
|
-
Requires-Dist: pillow
|
|
9
|
-
Requires-Dist: h5py
|
|
10
|
-
Provides-Extra: dev
|
|
11
|
-
Requires-Dist: pytest; extra == "dev"
|
|
12
|
-
Provides-Extra: gymnasium
|
|
13
|
-
Requires-Dist: gymnasium>=0.29.0; extra == "gymnasium"
|
|
14
|
-
Provides-Extra: video
|
|
15
|
-
Requires-Dist: moviepy>=2.1; extra == "video"
|
|
16
|
-
Provides-Extra: mjx
|
|
17
|
-
Requires-Dist: playground; extra == "mjx"
|
|
18
|
-
Provides-Extra: maniskill
|
|
19
|
-
Requires-Dist: mani_skill>=3.0.0b12; extra == "maniskill"
|
|
20
|
-
Dynamic: license-file
|
|
File without changes
|
|
File without changes
|
|
File without changes
|