From b1cfbac57046da02637687cb2c14493b8cfe9352 Mon Sep 17 00:00:00 2001 From: Michel Aractingi Date: Wed, 14 Jan 2026 17:09:56 +0100 Subject: [PATCH] remove type exp --- src/lerobot/scripts/lerobot_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py index 99ff41e43..d232f03da 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -104,7 +104,7 @@ def update_policy( if sample_weights is not None: # Use per-sample loss for weighted training # Note: Policies supporting sample weighting must implement forward(batch, reduction="none") - per_sample_loss, output_dict = policy.forward(batch, reduction="none") # type: ignore[call-arg] + per_sample_loss, output_dict = policy.forward(batch, reduction="none") # Apply sample weights: L_weighted = Σ(w_i * l_i) / (Σw_i + ε) # Weights are already normalized to sum to batch_size