Skip to content

Commit 993f0c7

Browse files
committed
Longer time for A100
1 parent 11df957 commit 993f0c7

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ci/L0_backend_vllm/metrics_test/vllm_metrics_test.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -127,14 +127,14 @@ def test_vllm_metrics(self):
127127

128128
# vllm:time_to_first_token_seconds
129129
self.assertEqual(metrics_dict["vllm:time_to_first_token_seconds_count"], 3)
130-
self.assertTrue(
131-
0 < metrics_dict["vllm:time_to_first_token_seconds_sum"] < 0.0005
132-
)
130+
self.assertTrue(0 < metrics_dict["vllm:time_to_first_token_seconds_sum"] < 0.01)
131+
self.assertEqual(metrics_dict["vllm:time_to_first_token_seconds_bucket"], 3)
133132
# vllm:time_per_output_token_seconds
134133
self.assertEqual(metrics_dict["vllm:time_per_output_token_seconds_count"], 45)
135134
self.assertTrue(
136-
0 <= metrics_dict["vllm:time_per_output_token_seconds_sum"] <= 0.005
135+
0 < metrics_dict["vllm:time_per_output_token_seconds_sum"] < 0.1
137136
)
137+
self.assertEqual(metrics_dict["vllm:time_per_output_token_seconds_bucket"], 45)
138138

139139
def test_vllm_metrics_disabled(self):
140140
# Test vLLM metrics

0 commit comments

Comments
 (0)