File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 2121from prometheus_client import Gauge
2222
2323from fastdeploy .metrics .metrics import get_filtered_metrics , main_process_metrics
24- from fastdeploy .spec_decode import SpecMethod
2524
2625
2726class 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 )
You can’t perform that action at this time.
0 commit comments