fix github bug

sinaptik-ai · Jan 8, 2025 · 2da1398 · 2da1398
1 parent bcec071
commit 2da1398
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extensions/llms/bedrock/tests/test_bedrock_claude.py b/extensions/llms/bedrock/tests/test_bedrock_claude.py
@@ -68,7 +68,7 @@ def test_params_setting(self):
         assert llm.top_k is None
         assert llm.max_tokens == 64
 
-    def test_call(self, mocker, prompt):
+    def test_call(self, prompt):
         llm = BedrockClaude(bedrock_runtime_client=MockBedrockRuntimeClient())
         expected_text = "This is the expected text."
         result = llm.call(instruction=prompt)