pyproject.toml

[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "himem-wam"
version = "0.1.0"
description = "HiMem-WAM: Hierarchical Memory-Gated World Action Models for robotic manipulation"
readme = "README.md"
requires-python = ">=3.10"
license = {text = "MIT"}
keywords = [
  "robotics",
  "imitation-learning",
  "world-action-model",
  "memory",
  "LIBERO",
]
dependencies = [
  "accelerate==1.12.0",
  "av==16.0.1",
  "boto3==1.35.99",
  "datasets==3.6.0",
  "einops==0.8.1",
  "gitpython==3.1.45",
  "huggingface-hub==0.29.2",
  "hydra-core==1.3.2",
  "imageio==2.37.0",
  "imageio-ffmpeg==0.6.0",
  "jsonlines==4.0.0",
  "modelscope==1.34.0",
  "numpy==1.26.4",
  "omegaconf==2.3.0",
  "packaging==25.0",
  "pandas==2.2.3",
  "pillow==12.0.0",
  "pyarrow==23.0.0",
  "pyyaml>=6.0",
  "regex==2025.11.3",
  "rich==14.2.0",
  "safetensors==0.5.3",
  "termcolor==2.5.0",
  "torch>=2.7.1",
  "torchcodec==0.5",
  "torchvision>=0.22.1",
  "tqdm==4.66.5",
  "transformers==4.49.0",
  "typing-extensions==4.15.0",
  "wandb==0.23.1",
]

[project.optional-dependencies]
dev = ["pytest>=8.0"]

[project.scripts]
himem-wam-infer-libero = "himem_wam.cli.infer_libero:main"

[tool.setuptools.packages.find]
include = ["himem_wam*"]