Fix vertexai_test break.

PiperOrigin-RevId: 688986487
google · Oct 23, 2024 · bddd008 · bddd008
1 parent e6a927d
commit bddd008
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/langfun/core/llms/vertexai_test.py b/langfun/core/llms/vertexai_test.py
@@ -195,8 +195,12 @@ def test_generation_config(self):
             stop=['\n'],
         ),
     )
+    actual = config.to_dict()
+    # There is a discrepancy between the `property_ordering` in the
+    # Google-internal version and the open-source version.
+    actual['response_schema'].pop('property_ordering', None)
     self.assertEqual(
-        config.to_dict(),
+        actual,
         dict(
             temperature=2.0,
             top_p=1.0,
@@ -209,7 +213,6 @@ def test_generation_config(self):
                 'properties': {
                     'name': {'type_': 'STRING'}
                 },
-                'property_ordering': ['name'],
                 'required': ['name'],
                 'title': 'Person',
             }