From 0264ac717bb9f98adc74852e26e1a4618201f453 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 6d6cea41e..d2355aaa9 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -101,7 +101,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