From 996f27a308b4a714f1259ace1f5e026b04fb62ba Mon Sep 17 00:00:00 2001
From: Rashmi Pawar <168514198+raspawar@users.noreply.github.com>
Date: Thu, 20 Feb 2025 21:56:47 +0530
Subject: [PATCH] fix: add logging import (#1174)

# What does this PR do?
Fixes logging import and the logger instance creation

cc: @dglogo
---
 llama_stack/providers/remote/inference/nvidia/nvidia.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/llama_stack/providers/remote/inference/nvidia/nvidia.py b/llama_stack/providers/remote/inference/nvidia/nvidia.py
index 0da617858..bcd29a0df 100644
--- a/llama_stack/providers/remote/inference/nvidia/nvidia.py
+++ b/llama_stack/providers/remote/inference/nvidia/nvidia.py
@@ -4,6 +4,7 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
+import logging
 import warnings
 from typing import AsyncIterator, List, Optional, Union
 
@@ -43,6 +44,8 @@ from .openai_utils import (
 )
 from .utils import _is_nvidia_hosted, check_health
 
+logger = logging.getLogger(__name__)
+
 
 class NVIDIAInferenceAdapter(Inference, ModelRegistryHelper):
     def __init__(self, config: NVIDIAConfig) -> None: