2024-02-29 02:11:29 +08:00
|
|
|
[tool.poetry]
|
|
|
|
|
name = "lerobot"
|
|
|
|
|
version = "0.1.0"
|
|
|
|
|
description = "Le robot is learning"
|
|
|
|
|
authors = [
|
|
|
|
|
"Rémi Cadène <re.cadene@gmail.com>",
|
|
|
|
|
"Simon Alibert <alibert.sim@gmail.com>",
|
|
|
|
|
]
|
|
|
|
|
repository = "https://github.com/Cadene/lerobot"
|
|
|
|
|
readme = "README.md"
|
|
|
|
|
license = "MIT"
|
|
|
|
|
classifiers=[
|
|
|
|
|
"Development Status :: 3 - Alpha",
|
|
|
|
|
"Intended Audience :: Developers",
|
|
|
|
|
"Topic :: Software Development :: Build Tools",
|
|
|
|
|
"License :: OSI Approved :: MIT License",
|
2024-03-05 23:03:03 +08:00
|
|
|
"Programming Language :: Python :: 3.10",
|
2024-02-29 02:11:29 +08:00
|
|
|
]
|
|
|
|
|
packages = [{include = "lerobot"}]
|
|
|
|
|
|
2024-03-11 19:10:46 +08:00
|
|
|
|
2024-02-29 02:11:29 +08:00
|
|
|
[tool.poetry.dependencies]
|
|
|
|
|
python = "^3.10"
|
|
|
|
|
termcolor = "^2.4.0"
|
|
|
|
|
omegaconf = "^2.3.0"
|
|
|
|
|
dm-env = "^1.6"
|
|
|
|
|
pandas = "^2.2.1"
|
|
|
|
|
wandb = "^0.16.3"
|
|
|
|
|
moviepy = "^1.0.3"
|
|
|
|
|
imageio = {extras = ["pyav"], version = "^2.34.0"}
|
|
|
|
|
gdown = "^5.1.0"
|
|
|
|
|
hydra-core = "^1.3.2"
|
|
|
|
|
einops = "^0.7.0"
|
|
|
|
|
pygame = "^2.5.2"
|
|
|
|
|
pymunk = "^6.6.0"
|
|
|
|
|
zarr = "^2.17.0"
|
|
|
|
|
shapely = "^2.0.3"
|
|
|
|
|
scikit-image = "^0.22.0"
|
|
|
|
|
numba = "^0.59.0"
|
|
|
|
|
mpmath = "^1.3.0"
|
|
|
|
|
torch = "^2.2.1"
|
|
|
|
|
tensordict = {git = "https://github.com/pytorch/tensordict"}
|
|
|
|
|
torchrl = {git = "https://github.com/pytorch/rl", rev = "13bef426dcfa5887c6e5034a6e9697993fa92c37"}
|
2024-03-26 00:25:55 +08:00
|
|
|
mujoco = "^2.3.7"
|
2024-02-29 02:11:29 +08:00
|
|
|
opencv-python = "^4.9.0.80"
|
2024-03-05 23:03:03 +08:00
|
|
|
diffusers = "^0.26.3"
|
|
|
|
|
torchvision = "^0.17.1"
|
2024-03-06 19:28:20 +08:00
|
|
|
h5py = "^3.10.0"
|
Add Aloha env and ACT policy
WIP Aloha env tests pass
Rendering works (fps look fast tho? TODO action bounding is too wide [-1,1])
Update README
Copy past from act repo
Remove download.py add a WIP for Simxarm
Remove download.py add a WIP for Simxarm
Add act yaml (TODO: try train.py)
Training can runs (TODO: eval)
Add tasks without end_effector that are compatible with dataset, Eval can run (TODO: training and pretrained model)
Add AbstractEnv, Refactor AlohaEnv, Add rendering_hook in env, Minor modifications, (TODO: Refactor Pusht and Simxarm)
poetry lock
fix bug in compute_stats for action normalization
fix more bugs in normalization
fix training
fix import
PushtEnv inheriates AbstractEnv, Improve factory Normalization
Add _make_env to EnvAbstract
Add call_rendering_hooks to pusht env
SimxarmEnv inherites from AbstractEnv (NOT TESTED)
Add aloha tests artifacts + update pusht stats
fix image normalization: before env was in [0,1] but dataset in [0,255], and now both in [0,255]
Small fix on simxarm
Add next to obs
Add top camera to Aloha env (TODO: make it compatible with set of cameras)
Add top camera to Aloha env (TODO: make it compatible with set of cameras)
2024-03-08 17:47:39 +08:00
|
|
|
dm-control = "1.0.14"
|
2024-03-15 08:30:11 +08:00
|
|
|
huggingface-hub = {extras = ["hf-transfer"], version = "^0.21.4"}
|
2024-03-21 18:18:50 +08:00
|
|
|
robomimic = "0.2.0"
|
2024-03-25 00:36:22 +08:00
|
|
|
gymnasium-robotics = "^1.2.4"
|
2024-03-25 22:12:21 +08:00
|
|
|
gymnasium = "^0.29.1"
|
2024-03-27 23:33:26 +08:00
|
|
|
cmake = "^3.29.0.1"
|
2024-03-31 23:05:25 +08:00
|
|
|
gym_pusht = { git = "git@github.com:huggingface/gym-pusht.git", optional = true}
|
2024-02-29 02:11:29 +08:00
|
|
|
|
2024-03-31 23:05:25 +08:00
|
|
|
[tool.poetry.extras]
|
|
|
|
|
pusht = ["gym_pusht"]
|
2024-02-29 02:11:29 +08:00
|
|
|
|
2024-02-29 19:26:35 +08:00
|
|
|
[tool.poetry.group.dev.dependencies]
|
|
|
|
|
pre-commit = "^3.6.2"
|
2024-03-05 23:03:03 +08:00
|
|
|
debugpy = "^1.8.1"
|
|
|
|
|
pytest = "^8.1.0"
|
2024-03-26 22:25:22 +08:00
|
|
|
pytest-cov = "^5.0.0"
|
2024-02-29 02:11:29 +08:00
|
|
|
|
2024-02-29 19:26:35 +08:00
|
|
|
|
|
|
|
|
[tool.ruff]
|
|
|
|
|
line-length = 110
|
|
|
|
|
target-version = "py310"
|
|
|
|
|
exclude = [
|
|
|
|
|
".bzr",
|
|
|
|
|
".direnv",
|
|
|
|
|
".eggs",
|
|
|
|
|
".git",
|
|
|
|
|
".git-rewrite",
|
|
|
|
|
".hg",
|
|
|
|
|
".mypy_cache",
|
|
|
|
|
".nox",
|
|
|
|
|
".pants.d",
|
|
|
|
|
".pytype",
|
|
|
|
|
".ruff_cache",
|
|
|
|
|
".svn",
|
|
|
|
|
".tox",
|
|
|
|
|
".venv",
|
|
|
|
|
"__pypackages__",
|
|
|
|
|
"_build",
|
|
|
|
|
"buck-out",
|
|
|
|
|
"build",
|
|
|
|
|
"dist",
|
|
|
|
|
"node_modules",
|
|
|
|
|
"venv",
|
|
|
|
|
]
|
|
|
|
|
|
2024-03-11 19:10:46 +08:00
|
|
|
|
2024-02-29 19:26:35 +08:00
|
|
|
[tool.ruff.lint]
|
|
|
|
|
select = ["E4", "E7", "E9", "F", "I", "N", "B", "C4", "SIM"]
|
2024-03-25 00:57:01 +08:00
|
|
|
ignore-init-module-imports = true
|
2024-03-11 19:10:46 +08:00
|
|
|
|
|
|
|
|
[tool.poetry-dynamic-versioning]
|
|
|
|
|
enable = true
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[build-system]
|
|
|
|
|
requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning>=1.0.0,<2.0.0"]
|
|
|
|
|
build-backend = "poetry_dynamic_versioning.backend"
|
