Extend lint test to test/ directory (#13247)

test/srt/cpu/test_activation.py

@@ -1,9 +1,7 @@
 import itertools
 import unittest
 
-import sgl_kernel
 import torch
-import torch.nn.functional as F
 from utils import GeluAndMul, SiluAndMul, precision
 
 from sglang.srt.server_args import ServerArgs, set_global_server_args_for_scheduler

test/srt/cpu/test_binding.py

@@ -1,7 +1,6 @@
 import re
 import unittest
 
-import sgl_kernel
 import torch
 
 kernel = torch.ops.sgl_kernel

test/srt/cpu/test_comm.py

@@ -5,11 +5,9 @@ import traceback
 import unittest
 from multiprocessing import Process
 
-import sgl_kernel
 import torch
 import torch.distributed as dist
 import torch.multiprocessing as mp
-from utils import precision
 
 from sglang.test.test_utils import CustomTestCase, find_available_port
 

test/srt/cpu/test_decode.py

@@ -1,6 +1,5 @@
 import unittest
 
-import sgl_kernel
 import torch
 from torch.nn.functional import scaled_dot_product_attention
 

test/srt/cpu/test_extend.py

@@ -1,6 +1,5 @@
 import unittest
 
-import sgl_kernel
 import torch
 from torch.nn.functional import scaled_dot_product_attention
 

test/srt/cpu/test_gemm.py

@@ -2,7 +2,6 @@ import itertools
 import unittest
 
 # TODO: use interface in cpu.py
-import sgl_kernel
 import torch
 import torch.nn as nn
 from utils import (

test/srt/cpu/test_mla.py

@@ -1,7 +1,5 @@
-import itertools
 import unittest
 
-import sgl_kernel
 import torch
 from torch.nn.functional import scaled_dot_product_attention
 from utils import precision

test/srt/cpu/test_moe.py

@@ -3,7 +3,6 @@ import math
 import unittest
 
 # TODO: use interface in cpu.py
-import sgl_kernel
 import torch
 
 kernel = torch.ops.sgl_kernel

test/srt/cpu/test_norm.py

@@ -2,7 +2,6 @@ import itertools
 import unittest
 from typing import Optional, Tuple, Union
 
-import sgl_kernel
 import torch
 from utils import make_non_contiguous, precision
 

test/srt/cpu/test_qkv_proj_with_rope.py

@@ -1,6 +1,5 @@
 import unittest
 
-import sgl_kernel
 import torch
 from utils import (
     convert_weight,

test/srt/cpu/test_rope.py

@@ -1,6 +1,5 @@
 import unittest
 
-import sgl_kernel
 import torch
 from utils import precision
 

test/srt/cpu/test_shared_expert.py

@@ -3,9 +3,7 @@ import math
 import unittest
 
 # TODO: use interface in cpu.py
-import sgl_kernel
 import torch
-import torch.nn as nn
 from utils import (
     BLOCK_K,
     BLOCK_N,

test/srt/cpu/test_topk.py

@@ -1,9 +1,6 @@
-import itertools
 import unittest
 
-import sgl_kernel
 import torch
-from utils import precision
 
 from sglang.srt.layers.moe.topk import (
     biased_grouped_topk_impl as native_biased_grouped_topk,