# HFIE stock Python 3.11 container ships torch + transformers + HF
# inference toolkit pre-baked. We override transformers to the version
# lerobot[pi]==0.5.1 was built against and pin the safe datasets release
# (4.8.x imports BucketNotFoundError which only exists on hub `main`).
#
# lerobot 0.5.1 declares requires-python>=3.12 in its metadata. The
# patch_lerobot_for_py311.py script in this repo rewrites the four
# PEP 695 sites so the code itself runs on 3.11, but pip's metadata
# check still fails by default. Set PIP_IGNORE_REQUIRES_PYTHON=1 in the
# endpoint's environment variables (HFIE config) to skip it.
#
# The shim package drops a .pth file into site-packages that re-exports
# the names the HF inference toolkit imports from transformers.file_utils
# (`is_tf_available`, `is_torch_available`). transformers 5.x dropped
# `is_tf_available` entirely (TF support removed). Without the shim the
# toolkit's webservice_starlette fails to start before our handler.py is
# ever loaded.
#
# Note: HFIE's pip install runs from a cwd OTHER than /repository, so
# `./shim` doesn't resolve. We use the absolute /repository/shim path
# (HFIE-specific: the runtime always mounts the model repo there).
/repository/shim
transformers==5.3.0
datasets==4.7.0
huggingface_hub>=0.36
lerobot==0.5.1