Merge branch 'main' into user/alexander-soare/train_pusht

This commit is contained in:
Alexander Soare
2024-03-12 19:41:27 +00:00
121 changed files with 3722 additions and 316 deletions

View File

@@ -18,6 +18,7 @@ classifiers=[
]
packages = [{include = "lerobot"}]
[tool.poetry.dependencies]
python = "^3.10"
cython = "^3.0.8"
@@ -41,7 +42,7 @@ mpmath = "^1.3.0"
torch = "^2.2.1"
tensordict = {git = "https://github.com/pytorch/tensordict"}
torchrl = {git = "https://github.com/pytorch/rl", rev = "13bef426dcfa5887c6e5034a6e9697993fa92c37"}
mujoco = "^3.1.2"
mujoco = "2.3.7"
mujoco-py = "^2.1.2.14"
gym = "^0.26.2"
opencv-python = "^4.9.0.80"
@@ -50,6 +51,7 @@ torchvision = "^0.17.1"
h5py = "^3.10.0"
robomimic = "0.2.0"
timm = "^0.9.16"
dm-control = "1.0.14"
[tool.poetry.group.dev.dependencies]
@@ -57,9 +59,6 @@ pre-commit = "^3.6.2"
debugpy = "^1.8.1"
pytest = "^8.1.0"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.ruff]
line-length = 110
@@ -88,5 +87,15 @@ exclude = [
"venv",
]
[tool.ruff.lint]
select = ["E4", "E7", "E9", "F", "I", "N", "B", "C4", "SIM"]
[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"