From 516376d27384d1f60e8313732e7c0babcf25be71 Mon Sep 17 00:00:00 2001 From: Will Dumm Date: Mon, 13 Jan 2025 14:10:50 -0800 Subject: [PATCH] format and lint --- tests/test_backward_compat.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/test_backward_compat.py b/tests/test_backward_compat.py index 708dbb41..27d382d7 100644 --- a/tests/test_backward_compat.py +++ b/tests/test_backward_compat.py @@ -11,7 +11,12 @@ def test_old_model_outputs(): dasm_crepe = load_crepe("tests/old_models/dasm_13k-v1jaffe+v1tang-joint") dnsm_crepe = load_crepe("tests/old_models/dnsm_13k-v1jaffe+v1tang-joint") - dasm_vals = torch.nan_to_num(set_wt_to_nan(torch.load("tests/old_models/dasm_output", weights_only=True), example_seq), 0.0) + dasm_vals = torch.nan_to_num( + set_wt_to_nan( + torch.load("tests/old_models/dasm_output", weights_only=True), example_seq + ), + 0.0, + ) dnsm_vals = torch.load("tests/old_models/dnsm_output", weights_only=True) dasm_result = torch.nan_to_num(dasm_crepe([example_seq])[0], 0.0)