Fix types

Change-Id: Id3e7316562f4029e5b7409ae725bb66e2207f075

Author: mayureshagashe2105

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(