From 00e9defb80ba06f1639a779450490bd056bf2a14 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Thu, 16 Apr 2026 20:21:58 +0200 Subject: [PATCH] fix(profiling): build flash-attn without isolation for groot groot depends on flash-attn, which fails to build in uv's default isolated build env because it doesn't declare torch as a build-time dependency. Torch is a core lerobot dep and is already present in the target venv when groot is synced, so we can safely disable build isolation just for flash-attn. The flag is a no-op for policies that don't pull in flash-attn. Made-with: Cursor --- .github/workflows/model_profiling.yml | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.github/workflows/model_profiling.yml b/.github/workflows/model_profiling.yml index a11718012..284879e3d 100644 --- a/.github/workflows/model_profiling.yml +++ b/.github/workflows/model_profiling.yml @@ -177,6 +177,11 @@ jobs: if [[ -n "${extra}" ]]; then sync_cmd+=(--extra "${extra}") fi + # flash-attn does not declare torch as a build-time dep, so its + # isolated build env fails with ModuleNotFoundError. Torch is a + # core lerobot dep and is already resolved here, so we disable + # build isolation for flash-attn specifically. + sync_cmd+=(--no-build-isolation-package flash-attn) if ! "${sync_cmd[@]}"; then echo "Dependency install failed for ${policy}; skipping." >&2 overall_status=1