fix: Remove duplicate import and make logger global. (#2817)

Signed-off-by: tzulingk@nvidia.com <tzulingk@nvidia.com>

tests/utils/deployment_graph.py

@@ -13,9 +13,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import logging
+import re
 from dataclasses import dataclass
 from typing import Any, Dict, List, Optional
 
+logger = logging.getLogger(__name__)
+
 
 @dataclass
 class Payload:
@@ -92,10 +96,11 @@ def completions_response_handler(response):
 
 def metrics_handler(response):
     """Handler to check if metrics endpoint is working and contains model label."""
-    import logging
-    import re
+    if response.status_code != 200:
+        raise AssertionError(
+            f"Metrics endpoint returned non-200 status code: {response.status_code}"
+        )
 
-    logger = logging.getLogger(__name__)
     metrics_text = response.text
 
     # Check for any model label in dynamo_component_requests_total metric

tests/utils/engine_process.py

@@ -34,7 +34,6 @@ class EngineProcess(ManagedProcess):
 
     def get_metrics(self, port=8081):
         """Curl the metrics endpoint and return the response."""
-        import requests
 
         metrics_url = f"http://localhost:{port}/metrics"
         logger.info(f"Curling metrics endpoint: {metrics_url}")