Skip to content

Commit d4981cd

Browse files
committed
Format README code examples
1 parent 458e14f commit d4981cd

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ import torch
138138
from vllm.config.compilation import CompilationConfig
139139
140140
141-
if __name__ == '__main__':
141+
if __name__ == "__main__":
142142
prompts = [
143143
"Hello, my name is",
144144
]

tests/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -303,6 +303,7 @@ Unit tests should be fast, isolated, and not require GPU or model weights. Use t
303303
import pytest
304304
import torch
305305
306+
306307
class TestMyOp:
307308
def test_basic(self, device):
308309
"""The `device` fixture returns cuda:0 or npu:0 automatically."""
@@ -322,6 +323,7 @@ import torch
322323

323324
pytestmark = pytest.mark.gpu
324325

326+
325327
def test_my_kernel(device):
326328
x = torch.randn(16, 256, device=device)
327329
result = my_kernel(x)

0 commit comments

Comments
 (0)