Formatting

tests/test_cli.py

@@ -121,13 +121,7 @@ def test_peft_training_params_are_fewer(policy_type, tmp_path):
     model_id = resolve_model_id_for_peft_training(policy_type)
 
     def dummy_update_policy(
-        train_metrics,
-        policy,
-        batch,
-        optimizer,
-        grad_clip_norm: float,
-        accelerator,
-        **kwargs
+        train_metrics, policy, batch, optimizer, grad_clip_norm: float, accelerator, **kwargs
     ):
         params_total = sum(p.numel() for p in policy.parameters())
         params_trainable = sum(p.numel() for p in policy.parameters() if p.requires_grad)