small fix to update diag

Author: Xuzzo

src/pydvl/influence/torch/influence_function_model.py

@@ -1153,7 +1153,7 @@ def _update_diag(
             hooks.append(module.register_forward_hook(input_hook))
             hooks.append(module.register_full_backward_hook(grad_hook))
 
-        for x, _ in tqdm(
+        for x, *_ in tqdm(
             data, disable=not self.progress, desc="Update Diagonal - batch progress"
         ):
             data_len += x.shape[0]