Skip to content

Commit e3c62a0

Browse files
committed
format
Change-Id: Ib3a0c90bfc6ec7f8f793917b3140769e2635a8e9
1 parent 5b825e3 commit e3c62a0

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/test_generative_models.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -811,7 +811,9 @@ def test_tool_config(self, tool_config, expected_tool_config):
811811
)
812812
def test_system_instruction(self, instruction, expected_instr):
813813
self.responses["generate_content"] = [simple_response("echo echo")]
814-
model = generative_models.GenerativeModel("gemini-1.5-flash", system_instruction=instruction)
814+
model = generative_models.GenerativeModel(
815+
"gemini-1.5-flash", system_instruction=instruction
816+
)
815817

816818
_ = model.generate_content("test")
817819

@@ -1206,7 +1208,9 @@ def test_repr_for_broken_streaming_chat(self):
12061208
self.assertEqual(expected, result)
12071209

12081210
def test_repr_for_system_instruction(self):
1209-
model = generative_models.GenerativeModel("gemini-1.5-flash", system_instruction="Be excellent.")
1211+
model = generative_models.GenerativeModel(
1212+
"gemini-1.5-flash", system_instruction="Be excellent."
1213+
)
12101214
result = repr(model)
12111215
self.assertIn("system_instruction='Be excellent.'", result)
12121216

0 commit comments

Comments
 (0)