Skip to content

Commit 82e7675

Browse files
committed
ruff fix
Signed-off-by: pavi2707 <2707pavithra@gmail.com>
1 parent cc1d856 commit 82e7675

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

tests/models/test_decoders.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -438,7 +438,12 @@ def persistent_model():
438438
"model_path,batch_size,seq_length,max_new_tokens", common_shapes
439439
)
440440
def test_common_shapes(
441-
model_path, batch_size, seq_length, max_new_tokens, persistent_model, record_property
441+
model_path,
442+
batch_size,
443+
seq_length,
444+
max_new_tokens,
445+
persistent_model,
446+
record_property,
442447
):
443448
torch.manual_seed(42)
444449
torch.set_grad_enabled(False)
@@ -684,7 +689,7 @@ def _metric_calculator(r: torch.Tensor, t: torch.Tensor):
684689
dprint(f"mean diff failure rate: {diff_failure_rate}")
685690
dprint(f"cross entropy loss failure rate: {ce_failure_rate}")
686691
# Add failure rates to xml report
687-
record_property("mean_diff_failure_rate", diff_failure_rate)
692+
record_property("mean_diff_failure_rate", diff_failure_rate)
688693
record_property("cross_entropy_loss_failure_rate", ce_failure_rate)
689694
if "mean_diff" not in skip_assertions:
690695
assert diff_failure_rate < failure_rate_threshold, (

0 commit comments

Comments
 (0)