diff --git a/tests/compile/piecewise/test_simple.py b/tests/compile/piecewise/test_simple.py index 51bab00739da7..a1535ff6c5ebc 100644 --- a/tests/compile/piecewise/test_simple.py +++ b/tests/compile/piecewise/test_simple.py @@ -103,4 +103,4 @@ def test_simple_piecewise_compile(use_inductor): batch_descriptor=BatchDescriptor(num_tokens=2, )): output = model(input) assert get_global_counter() == 2 - assert torch.allclose(output.cpu(), torch.tensor([19., 19.])) \ No newline at end of file + assert torch.allclose(output.cpu(), torch.tensor([19., 19.]))