Skip to content

Commit fc2bcf4

Browse files
committed
[fix] fix test_metrics
1 parent 8cdfa1e commit fc2bcf4

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

tests/metrics/test_metrics.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
from prometheus_client import Gauge
2222

2323
from fastdeploy.metrics.metrics import get_filtered_metrics, main_process_metrics
24-
from fastdeploy.spec_decode import SpecMethod
2524

2625

2726
class TestGetFilteredMetrics(unittest.TestCase):
@@ -54,7 +53,7 @@ def test_regular_gauge_returns_single_value_without_pid(self):
5453

5554
def test_speculative_gauge_returns_single_value_without_pid(self):
5655
if not hasattr(main_process_metrics, "spec_decode_draft_acceptance_rate"):
57-
main_process_metrics._init_speculative_metrics(SpecMethod.MTP, 2)
56+
main_process_metrics._init_speculative_metrics("mtp", 2)
5857

5958
metric = main_process_metrics.spec_decode_draft_acceptance_rate
6059
metric.set(0.75)
@@ -65,7 +64,7 @@ def test_speculative_gauge_returns_single_value_without_pid(self):
6564

6665
def test_speculative_single_head_gauge_returns_single_value_without_pid(self):
6766
if not hasattr(main_process_metrics, "spec_decode_draft_acceptance_rate"):
68-
main_process_metrics._init_speculative_metrics(SpecMethod.MTP, 2)
67+
main_process_metrics._init_speculative_metrics("mtp", 2)
6968

7069
metric = main_process_metrics.spec_decode_draft_single_head_acceptance_rate[0]
7170
metric.set(0.6)

0 commit comments

Comments
 (0)