Skip to content

Commit 8a2507b

Browse files
committed
temp: remove pragma no cover comments
1 parent d751872 commit 8a2507b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

nemoguardrails/llm/providers/_langchain_nvidia_ai_endpoints_patch.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,10 +32,10 @@
3232
from langchain_nvidia_ai_endpoints import ChatNVIDIA as ChatNVIDIAOriginal
3333
from pydantic import Field
3434

35-
log = logging.getLogger(__name__) # pragma: no cover
35+
log = logging.getLogger(__name__)
3636

3737

38-
def stream_decorator(func): # pragma: no cover
38+
def stream_decorator(func):
3939
@wraps(func)
4040
def wrapper(
4141
self,
@@ -81,7 +81,7 @@ async def wrapper(
8181

8282
# NOTE: this needs to have the same name as the original class,
8383
# otherwise, there's a check inside `langchain-nvidia-ai-endpoints` that will fail.
84-
class ChatNVIDIA(ChatNVIDIAOriginal): # pragma: no cover
84+
class ChatNVIDIA(ChatNVIDIAOriginal):
8585
streaming: bool = Field(
8686
default=False, description="Whether to use streaming or not"
8787
)

0 commit comments

Comments
 (0)