pylint..

log2timeline · Dec 22, 2024 · c327b87 · c327b87
1 parent d539357
commit c327b87
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/lib/processors/llmproviders/gemini.py b/tests/lib/processors/llmproviders/gemini.py
@@ -123,9 +123,7 @@ def test_generate_with_history(self, mock_gen_model, mock_gen_config):
       values={'GOOGLE_API_KEY': 'fake_env_key'},
       clear=True
   )
-  @mock.patch('google.generativeai.configure')
-  @mock.patch('google.generativeai.GenerativeModel', autospec=True)
-  def test_generate_from_template(self, mock_gen_model, mock_config):
+  def test_generate_from_template(self):
     """Tests the GenerateFromTemplate method."""
     provider = gemini.GeminiLLMProvider()
     mock_generate = mock.Mock()