diff --git a/tests/compile/piecewise/test_multiple_graphs.py b/tests/compile/piecewise/test_multiple_graphs.py index 1f46ca1f9cfe8..967167e2d160d 100644 --- a/tests/compile/piecewise/test_multiple_graphs.py +++ b/tests/compile/piecewise/test_multiple_graphs.py @@ -320,4 +320,4 @@ def test_multi_graph_piecewise_compile_outputs_equal(): outputs.append(run_model(vllm_config, model, inputs)) # Expect bitwise equivalence using inductor w/ and w/o cudagraph - assert torch.equal(outputs[0], outputs[2]) \ No newline at end of file + assert torch.equal(outputs[0], outputs[2])