546719137a
Added tensordict dependencies Updated the version of torch and torchvision Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
135 lines
4.3 KiB
TOML
135 lines
4.3 KiB
TOML
[tool.poetry]
|
|
name = "lerobot"
|
|
version = "0.1.0"
|
|
description = "🤗 LeRobot: State-of-the-art Machine Learning for Real-World Robotics in Pytorch"
|
|
authors = [
|
|
"Rémi Cadène <re.cadene@gmail.com>",
|
|
"Simon Alibert <alibert.sim@gmail.com>",
|
|
"Alexander Soare <alexander.soare159@gmail.com>",
|
|
"Quentin Gallouédec <quentin.gallouedec@ec-lyon.fr>",
|
|
"Adil Zouitine <adilzouitinegm@gmail.com>",
|
|
"Thomas Wolf <thomaswolfcontact@gmail.com>",
|
|
]
|
|
repository = "https://github.com/huggingface/lerobot"
|
|
readme = "README.md"
|
|
license = "Apache-2.0"
|
|
classifiers=[
|
|
"Development Status :: 3 - Alpha",
|
|
"Intended Audience :: Developers",
|
|
"Intended Audience :: Education",
|
|
"Intended Audience :: Science/Research",
|
|
"Topic :: Software Development :: Build Tools",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"License :: OSI Approved :: Apache Software License",
|
|
"Programming Language :: Python :: 3.10",
|
|
]
|
|
packages = [{include = "lerobot"}]
|
|
|
|
|
|
[tool.poetry.dependencies]
|
|
python = ">=3.10,<3.13"
|
|
termcolor = ">=2.4.0"
|
|
omegaconf = ">=2.3.0"
|
|
wandb = ">=0.16.3"
|
|
imageio = {extras = ["ffmpeg"], version = ">=2.34.0"}
|
|
gdown = ">=5.1.0"
|
|
hydra-core = ">=1.3.2"
|
|
einops = ">=0.8.0"
|
|
pymunk = ">=6.6.0"
|
|
zarr = ">=2.17.0"
|
|
numba = ">=0.59.0"
|
|
torch = ">=2.6.0"
|
|
opencv-python = ">=4.9.0"
|
|
diffusers = ">=0.27.2"
|
|
torchvision = ">=0.21.0"
|
|
h5py = ">=3.10.0"
|
|
huggingface-hub = {extras = ["hf-transfer", "cli"], version = ">=0.25.2"}
|
|
gymnasium = "==0.29.1" # TODO(rcadene, aliberts): Make gym 1.0.0 work
|
|
cmake = ">=3.29.0.1"
|
|
gym-dora = { git = "https://github.com/dora-rs/dora-lerobot.git", subdirectory = "gym_dora", optional = true }
|
|
gym-pusht = { version = ">=0.1.5", optional = true}
|
|
gym-xarm = { version = ">=0.1.1", optional = true}
|
|
gym-aloha = { version = ">=0.1.1", optional = true}
|
|
pre-commit = {version = ">=3.7.0", optional = true}
|
|
debugpy = {version = ">=1.8.1", optional = true}
|
|
pytest = {version = ">=8.1.0", optional = true}
|
|
pytest-cov = {version = ">=5.0.0", optional = true}
|
|
datasets = ">=2.19.0"
|
|
imagecodecs = { version = ">=2024.1.1", optional = true }
|
|
pyav = ">=12.0.5"
|
|
rerun-sdk = ">=0.15.1"
|
|
deepdiff = ">=7.0.1"
|
|
flask = ">=3.0.3"
|
|
pandas = {version = ">=2.2.2", optional = true}
|
|
scikit-image = {version = ">=0.23.2", optional = true}
|
|
dynamixel-sdk = {version = ">=3.7.31", optional = true}
|
|
pynput = {version = ">=1.7.7", optional = true}
|
|
feetech-servo-sdk = {version = ">=1.0.0", optional = true}
|
|
setuptools = {version = "!=71.0.1", optional = true} # TODO(rcadene, aliberts): 71.0.1 has a bug
|
|
pyrealsense2 = {version = ">=2.55.1.6486", markers = "sys_platform != 'darwin'", optional = true} # TODO(rcadene, aliberts): Fix on Mac
|
|
pyrender = {git = "https://github.com/mmatl/pyrender.git", markers = "sys_platform == 'linux'", optional = true}
|
|
hello-robot-stretch-body = {version = ">=0.7.27", markers = "sys_platform == 'linux'", optional = true}
|
|
pyserial = {version = ">=3.5", optional = true}
|
|
jsonlines = ">=4.0.0"
|
|
transformers = {version = ">=4.47.0", optional = true}
|
|
torchmetrics = {version = ">=1.6.0", optional = true}
|
|
grpcio = { version = "^1.70.0", optional = true }
|
|
protobuf = { version = "^5.29.3", optional = true }
|
|
mani-skill = { version = "^3.0.0b18", optional = true }
|
|
|
|
|
|
[tool.poetry.extras]
|
|
dora = ["gym-dora"]
|
|
pusht = ["gym-pusht"]
|
|
xarm = ["gym-xarm"]
|
|
aloha = ["gym-aloha"]
|
|
dev = ["pre-commit", "debugpy"]
|
|
test = ["pytest", "pytest-cov", "pyserial"]
|
|
umi = ["imagecodecs"]
|
|
video_benchmark = ["scikit-image", "pandas"]
|
|
dynamixel = ["dynamixel-sdk", "pynput"]
|
|
feetech = ["feetech-servo-sdk", "pynput"]
|
|
intelrealsense = ["pyrealsense2"]
|
|
stretch = ["hello-robot-stretch-body", "pyrender", "pyrealsense2", "pynput"]
|
|
hilserl = ["transformers", "torchmetrics", "grpcio", "protobuf", "tensordict"]
|
|
mani_skill = ["mani-skill"]
|
|
|
|
[tool.ruff]
|
|
line-length = 110
|
|
target-version = "py310"
|
|
exclude = [
|
|
"tests/data",
|
|
".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",
|
|
"*_pb2.py",
|
|
"*_pb2_grpc.py",
|
|
]
|
|
|
|
|
|
[tool.ruff.lint]
|
|
select = ["E4", "E7", "E9", "F", "I", "N", "B", "C4", "SIM"]
|
|
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|