format

Change-Id: Ib3a0c90bfc6ec7f8f793917b3140769e2635a8e9
google-gemini · Oct 15, 2024 · e3c62a0 · e3c62a0
1 parent 5b825e3
commit e3c62a0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test_generative_models.py b/tests/test_generative_models.py
@@ -811,7 +811,9 @@ def test_tool_config(self, tool_config, expected_tool_config):
     )
     def test_system_instruction(self, instruction, expected_instr):
         self.responses["generate_content"] = [simple_response("echo echo")]
-        model = generative_models.GenerativeModel("gemini-1.5-flash", system_instruction=instruction)
+        model = generative_models.GenerativeModel(
+            "gemini-1.5-flash", system_instruction=instruction
+        )
 
         _ = model.generate_content("test")
 
@@ -1206,7 +1208,9 @@ def test_repr_for_broken_streaming_chat(self):
         self.assertEqual(expected, result)
 
     def test_repr_for_system_instruction(self):
-        model = generative_models.GenerativeModel("gemini-1.5-flash", system_instruction="Be excellent.")
+        model = generative_models.GenerativeModel(
+            "gemini-1.5-flash", system_instruction="Be excellent."
+        )
         result = repr(model)
         self.assertIn("system_instruction='Be excellent.'", result)