From bfbaf09fa8baeb5a88e2f4681bf4438e6243b71f Mon Sep 17 00:00:00 2001 From: Jash Gulabrai Date: Wed, 30 Apr 2025 10:57:40 -0400 Subject: [PATCH] fix unit tests --- tests/unit/providers/nvidia/test_safety.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/providers/nvidia/test_safety.py b/tests/unit/providers/nvidia/test_safety.py index a237c56b0..8c74f178b 100644 --- a/tests/unit/providers/nvidia/test_safety.py +++ b/tests/unit/providers/nvidia/test_safety.py @@ -139,7 +139,7 @@ class TestNVIDIASafetyAdapter(unittest.TestCase): "model": shield_id, "messages": [ {"role": "user", "content": "Hello, how are you?"}, - {"role": "assistant", "content": "I'm doing well, thank you for asking!", "tool_calls": None}, + {"role": "assistant", "content": "I'm doing well, thank you for asking!"}, ], "temperature": 1.0, "top_p": 1, @@ -193,7 +193,7 @@ class TestNVIDIASafetyAdapter(unittest.TestCase): "model": shield_id, "messages": [ {"role": "user", "content": "Hello, how are you?"}, - {"role": "assistant", "content": "I'm doing well, thank you for asking!", "tool_calls": None}, + {"role": "assistant", "content": "I'm doing well, thank you for asking!"}, ], "temperature": 1.0, "top_p": 1, @@ -269,7 +269,7 @@ class TestNVIDIASafetyAdapter(unittest.TestCase): "model": shield_id, "messages": [ {"role": "user", "content": "Hello, how are you?"}, - {"role": "assistant", "content": "I'm doing well, thank you for asking!", "tool_calls": None}, + {"role": "assistant", "content": "I'm doing well, thank you for asking!"}, ], "temperature": 1.0, "top_p": 1,