107 lines
2.7 KiB
TOML
107 lines
2.7 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>",
|
|
"Alexander Soare <alexander.soare159@gmail.com>",
|
|
"Quentin Gallouédec <quentin.gallouedec@ec-lyon.fr>",
|
|
"Simon Alibert <alibert.sim@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"
|
|
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.7.0"
|
|
pymunk = "^6.6.0"
|
|
zarr = "^2.17.0"
|
|
numba = "^0.59.0"
|
|
torch = "^2.2.1"
|
|
opencv-python = "^4.9.0.80"
|
|
diffusers = "^0.26.3"
|
|
torchvision = "^0.18.0"
|
|
h5py = "^3.10.0"
|
|
huggingface-hub = "^0.21.4"
|
|
robomimic = "0.2.0"
|
|
gymnasium = "^0.29.1"
|
|
cmake = "^3.29.0.1"
|
|
gym-pusht = { git = "git@github.com:huggingface/gym-pusht.git", optional = true}
|
|
gym-xarm = { git = "git@github.com:huggingface/gym-xarm.git", optional = true}
|
|
gym-aloha = { git = "git@github.com:huggingface/gym-aloha.git", 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 }
|
|
torchaudio = "^2.3.0"
|
|
|
|
|
|
[tool.poetry.extras]
|
|
pusht = ["gym-pusht"]
|
|
xarm = ["gym-xarm"]
|
|
aloha = ["gym-aloha"]
|
|
dev = ["pre-commit", "debugpy"]
|
|
test = ["pytest", "pytest-cov"]
|
|
umi = ["imagecodecs"]
|
|
|
|
|
|
[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",
|
|
]
|
|
|
|
|
|
[tool.ruff.lint]
|
|
select = ["E4", "E7", "E9", "F", "I", "N", "B", "C4", "SIM"]
|
|
ignore-init-module-imports = true
|
|
|
|
|
|
[build-system]
|
|
requires = ["poetry-core>=1.5.0"]
|
|
build-backend = "poetry.core.masonry.api"
|