accelerate<2.0.0,>=1.10.0
av<16.0.0,>=15.0.0
datasets<5.0.0,>=4.0.0
deepdiff<9.0.0,>=7.0.1
diffusers<0.36.0,>=0.27.2
draccus==0.10.0
einops<0.9.0,>=0.8.0
gymnasium<2.0.0,>=1.1.1
h5py>=3.1.0
huggingface-hub<2.0.0,>=1.0.0
hydra-core>=1.3.0
imageio[ffmpeg]<3.0.0,>=2.34.0
jsonlines<5.0.0,>=4.0.0
matplotlib>=3.3.0
numpy<2.3.0,>=2.0.0
num2words<1.0.0,>=0.5.14
opencv-python-headless<4.14.0,>=4.9.0
omegaconf>=2.3.0
packaging<26.0,>=24.2
pyserial>=3.5
requests>=2.31.0
rerun-sdk<0.27.0,>=0.24.0
PyYAML>=6.0.0
safetensors<1.0.0,>=0.4.3
scipy>=1.10.0
shutup>=0.2.0
tensorboard>=2.0.0
termcolor<4.0.0,>=2.4.0
torch<2.11.0,>=2.2.1
torchvision<0.26.0,>=0.21.0
lerobot[smolvla]==0.5.1
transformers==5.3.0
tqdm>=4.66.0
wandb<0.25.0,>=0.24.0

[all]
gym-aloha<0.2.0,>=0.1.2
gym-pusht<0.2.0,>=0.1.5
pymunk<7.0.0,>=6.6.0
metaworld==3.0.0
pytest>=8.1.0
pre-commit<5.0.0,>=3.7.0
debugpy<1.9.0,>=1.8.1
mypy>=1.19.1
lerobot[pi,wallx,xvla]==0.5.1
scipy<2.0.0,>=1.14.0
sentencepiece>=0.1.99
tiktoken>=0.6.0
peft<1.0.0,>=0.18.0
torchdiffeq<0.3.0,>=0.2.4
qwen-vl-utils<0.1.0,>=0.0.11

[all:sys_platform == "linux"]
hf-libero<0.2.0,>=0.1.3

[aloha]
gym-aloha<0.2.0,>=0.1.2
scipy<2.0.0,>=1.14.0

[dev]
pytest>=8.1.0
pre-commit<5.0.0,>=3.7.0
debugpy<1.9.0,>=1.8.1
mypy>=1.19.1

[envs]
gym-aloha<0.2.0,>=0.1.2
gym-pusht<0.2.0,>=0.1.5
pymunk<7.0.0,>=6.6.0
metaworld==3.0.0

[envs:sys_platform == "linux"]
hf-libero<0.2.0,>=0.1.3

[groot]
lerobot[groot]==0.5.1
peft<1.0.0,>=0.18.0
dm-tree<1.0.0,>=0.1.8
timm<1.1.0,>=1.0.0
Pillow<13.0.0,>=10.0.0
ninja<2.0.0,>=1.11.1

[groot:sys_platform != "darwin"]
flash-attn<3.0.0,>=2.5.9

[libero]
scipy<2.0.0,>=1.14.0

[libero:sys_platform == "linux"]
hf-libero<0.2.0,>=0.1.3

[metaworld]
metaworld==3.0.0
scipy<2.0.0,>=1.14.0

[pi]
lerobot[pi]==0.5.1
sentencepiece>=0.1.99
tiktoken>=0.6.0

[pusht]
gym-pusht<0.2.0,>=0.1.5
pymunk<7.0.0,>=6.6.0

[vla_policies]
lerobot[pi,wallx,xvla]==0.5.1
scipy<2.0.0,>=1.14.0
sentencepiece>=0.1.99
tiktoken>=0.6.0
peft<1.0.0,>=0.18.0
torchdiffeq<0.3.0,>=0.2.4
qwen-vl-utils<0.1.0,>=0.0.11

[wallx]
lerobot[wallx]==0.5.1
peft<1.0.0,>=0.18.0
scipy<2.0.0,>=1.14.0
torchdiffeq<0.3.0,>=0.2.4
qwen-vl-utils<0.1.0,>=0.0.11

[xvla]
lerobot[xvla]==0.5.1
