Correct test location

Signed-off-by: Seiji Eicher <seiji@anyscale.com>

Author: eicherseiji

tests/metrics/test_ray_metrics.py renamed to tests/v1/metrics/test_ray_metrics.py

@@ -29,6 +29,8 @@ def test_engine_log_metrics_ray(
     dtype: str,
     max_tokens: int,
 ) -> None:
+    """ Simple smoke test, verifying this can be used without exceptions.
+    Need to start a Ray cluster in order to verify outputs."""
 
     @ray.remote(num_gpus=1)
     class EngineTestActor:

vllm/v1/metrics/ray_wrappers.py

@@ -2,8 +2,6 @@
 import time
 from typing import Optional, Union
 
-import prometheus_client
-
 from vllm.config import VllmConfig
 from vllm.v1.metrics.loggers import PrometheusStatLogger
 from vllm.v1.spec_decode.metrics import SpecDecodingProm
@@ -102,7 +100,7 @@ class RaySpecDecodingProm(SpecDecodingProm):
     util.metrics library.
     """
 
-    _counter_cls = prometheus_client.Counter
+    _counter_cls = RayCounterWrapper
 
 
 class RayPrometheusStatLogger(PrometheusStatLogger):