Fix types

Change-Id: Id3e7316562f4029e5b7409ae725bb66e2207f075
mayureshagashe2105 · May 27, 2024 · 67472d3 · 67472d3
1 parent a1c8c72
commit 67472d3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_generative_models.py b/tests/test_generative_models.py
@@ -355,8 +355,8 @@ def test_model_with_cached_content_as_context(self, cached_content):
         self.assertEqual(cc_name, "cachedContents/test-cached-content")
         self.assertEqual(model_name, "models/gemini-1.0-pro-001")
         self.assertEqual(
-            model.cached_content, "cachedContents/test-cached-content"
-        )  # pytype: disable=attribute-error
+            model.cached_content, "cachedContents/test-cached-content"  # pytype: disable=attribute-error
+        )
 
     def test_content_generation_with_model_having_context(self):
         self.responses["generate_content"] = [simple_response("world!")]
@@ -367,8 +367,8 @@ def test_content_generation_with_model_having_context(self):
 
         self.assertEqual(response.text, "world!")
         self.assertEqual(
-            model.cached_content, "cachedContents/test-cached-content"
-        )  # pytype: disable=attribute-error
+            model.cached_content, "cachedContents/test-cached-content"  # pytype: disable=attribute-error
+        )
 
     def test_fail_content_generation_with_model_having_context(self):
         model = generative_models.GenerativeModel.from_cached_content(