update pre-commit config (#18860)

This commit is contained in:
SoluMilken
2026-02-16 00:18:31 +08:00
committed by GitHub
parent f7603203b0
commit 07a24f1a38
135 changed files with 239 additions and 198 deletions
@@ -43,12 +43,9 @@ class TestSlidingWindowAttentionTriton(CustomTestCase):
cls.short_context_prompt = "The capital of France is"
# Test prompt longer than window size
cls.long_context_prompt = (
"""
cls.long_context_prompt = """
Once upon a time, there was a mountain. In the mountain, there was a temple. In the temple, there was an old monk telling a story. The story was:
"""
* 100
)
""" * 100
cls.long_context_prompt += "\nNow, summarize the story in one sentence:"
def _test_mmlu(self):
@@ -10,7 +10,10 @@ import torch.nn.functional as F
from sglang.srt.layers.attention.fla.layernorm_gated import (
_layer_norm_fwd as layer_norm_fwd,
)
from sglang.srt.layers.attention.fla.layernorm_gated import layernorm_fn, rms_norm_ref
from sglang.srt.layers.attention.fla.layernorm_gated import (
layernorm_fn,
rms_norm_ref,
)
from sglang.test.ci.ci_register import register_cuda_ci
register_cuda_ci(
@@ -7,6 +7,7 @@ unittest version of the AWQ Triton kernel tests.
Run with:
python -m unittest test_awq_dequant.py
"""
import unittest
import torch