[[source]]
url = "https://pypi.org/simple"
verify_ssl = true
name = "pypi"

[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu121"
verify_ssl = false

[packages]
torch = { index = "pytorch", version = ">=2.2.0, <2.3.0" }
torchvision = { index = "pytorch", version = ">=0.16.0" }
# libero = { file = "LIBERO", editable = true }
openvla = { file = "openvla", editable = true }
imageio = { extras = ["ffmpeg"], version = "*" }
draccus = "==0.8.0"
tqdm = "*"
wandb = "==0.13.1"
bddl = "==1.0.1"
easydict = "==1.9"
cloudpickle = "==2.1.0"
gym = "==0.25.2"
timm = "==0.9.10"
transformers = "==4.40.1"
hydra-core = "==1.2.0"
numpy = "<2.0.0"
opencv-python = "==4.6.0.66"
robomimic = "==0.2.0"
einops = "==0.4.1"
thop = "==0.1.1-2209072238"
future = "==0.18.2"
matplotlib = "==3.5.3"
setuptools = "*"
dlimp = { git = "git+https://github.com/moojink/dlimp_openvla" }
tensorflow = "*"
tensorflow-graphics = "*"
accelerate = "*"
pyyaml = "*"
openai = "*"
spacy = "*"
uv = "*"

[dev-packages]

[scripts]
post_install = "pip install flash-attn==2.5.5 --no-build-isolation"
pre_install = "pip install --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host files.pythonhosted.org robosuite==1.4.1"

[requires]
python_version = "3.10"
