Migrate metrics tests to test/registered/metrics/ (#16466)
This commit is contained in:
@@ -2,6 +2,11 @@ import unittest
|
||||
from typing import Dict, List
|
||||
|
||||
import requests
|
||||
|
||||
from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
|
||||
|
||||
register_cuda_ci(est_time=32, suite="stage-b-test-small-1-gpu")
|
||||
register_amd_ci(est_time=32, suite="stage-b-test-small-1-gpu")
|
||||
from prometheus_client.parser import text_string_to_metric_families
|
||||
from prometheus_client.samples import Sample
|
||||
|
||||
@@ -1,6 +1,9 @@
|
||||
import unittest
|
||||
|
||||
from sglang.srt.metrics.utils import generate_buckets, two_sides_exponential_buckets
|
||||
from sglang.test.ci.ci_register import register_cpu_ci
|
||||
|
||||
register_cpu_ci(est_time=1, suite="stage-a-cpu-only")
|
||||
|
||||
|
||||
class TestMetricsUtils(unittest.TestCase):
|
||||
@@ -44,8 +44,6 @@ suites = {
|
||||
TestFile("test_io_struct.py", 8),
|
||||
TestFile("test_jinja_template_utils.py", 7),
|
||||
TestFile("test_mamba_unittest.py", 9),
|
||||
TestFile("test_metrics.py", 32),
|
||||
TestFile("test_metrics_utils.py", 1),
|
||||
TestFile("test_model_hooks.py", 6),
|
||||
TestFile("test_modelopt_loader.py", 11),
|
||||
TestFile("test_multi_tokenizer.py", 230),
|
||||
@@ -190,8 +188,6 @@ suite_amd = {
|
||||
TestFile("test_input_embeddings.py", 38),
|
||||
TestFile("test_io_struct.py", 8),
|
||||
TestFile("test_jinja_template_utils.py", 1),
|
||||
TestFile("test_metrics.py", 32),
|
||||
TestFile("test_metrics_utils.py", 1),
|
||||
TestFile("test_model_hooks.py", 10),
|
||||
TestFile("test_multi_tokenizer.py", 345),
|
||||
TestFile("test_no_chunked_prefill.py", 108),
|
||||
|
||||
Reference in New Issue
Block a user