Migrate metrics tests to test/registered/metrics/ (#16466)
This commit is contained in:
@@ -2,6 +2,11 @@ import unittest
|
|||||||
from typing import Dict, List
|
from typing import Dict, List
|
||||||
|
|
||||||
import requests
|
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.parser import text_string_to_metric_families
|
||||||
from prometheus_client.samples import Sample
|
from prometheus_client.samples import Sample
|
||||||
|
|
||||||
@@ -1,6 +1,9 @@
|
|||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from sglang.srt.metrics.utils import generate_buckets, two_sides_exponential_buckets
|
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):
|
class TestMetricsUtils(unittest.TestCase):
|
||||||
@@ -44,8 +44,6 @@ suites = {
|
|||||||
TestFile("test_io_struct.py", 8),
|
TestFile("test_io_struct.py", 8),
|
||||||
TestFile("test_jinja_template_utils.py", 7),
|
TestFile("test_jinja_template_utils.py", 7),
|
||||||
TestFile("test_mamba_unittest.py", 9),
|
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_model_hooks.py", 6),
|
||||||
TestFile("test_modelopt_loader.py", 11),
|
TestFile("test_modelopt_loader.py", 11),
|
||||||
TestFile("test_multi_tokenizer.py", 230),
|
TestFile("test_multi_tokenizer.py", 230),
|
||||||
@@ -190,8 +188,6 @@ suite_amd = {
|
|||||||
TestFile("test_input_embeddings.py", 38),
|
TestFile("test_input_embeddings.py", 38),
|
||||||
TestFile("test_io_struct.py", 8),
|
TestFile("test_io_struct.py", 8),
|
||||||
TestFile("test_jinja_template_utils.py", 1),
|
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_model_hooks.py", 10),
|
||||||
TestFile("test_multi_tokenizer.py", 345),
|
TestFile("test_multi_tokenizer.py", 345),
|
||||||
TestFile("test_no_chunked_prefill.py", 108),
|
TestFile("test_no_chunked_prefill.py", 108),
|
||||||
|
|||||||
Reference in New Issue
Block a user