fix: Use == instead of .__eq__ in rag_corpus_eq\n\nReplaced .__eq__() with == for object comparisons in the rag_corpus_eq helper function to resolve TypeError in Python 3.14.

Author: chalmerlowe

tests/unit/vertex_rag/test_rag_data.py

@@ -419,10 +419,8 @@ def create_transformation_config(
 def rag_corpus_eq(returned_corpus, expected_corpus):
     assert returned_corpus.name == expected_corpus.name
     assert returned_corpus.display_name == expected_corpus.display_name
-    assert returned_corpus.backend_config.__eq__(expected_corpus.backend_config)
-    assert returned_corpus.vertex_ai_search_config.__eq__(
-        expected_corpus.vertex_ai_search_config
-    )
+    assert returned_corpus.backend_config == expected_corpus.backend_config
+    assert returned_corpus.vertex_ai_search_config == expected_corpus.vertex_ai_search_config
 
 
 def rag_file_eq(returned_file, expected_file):

tests/unit/vertex_rag/test_rag_data_preview.py

@@ -691,12 +691,10 @@ def create_transformation_config(
 def rag_corpus_eq(returned_corpus, expected_corpus):
     assert returned_corpus.name == expected_corpus.name
     assert returned_corpus.display_name == expected_corpus.display_name
-    assert returned_corpus.vector_db.__eq__(expected_corpus.vector_db)
-    assert returned_corpus.backend_config.__eq__(expected_corpus.backend_config)
-    assert returned_corpus.vertex_ai_search_config.__eq__(
-        expected_corpus.vertex_ai_search_config
-    )
-    assert returned_corpus.corpus_type_config.__eq__(expected_corpus.corpus_type_config)
+    assert returned_corpus.vector_db == expected_corpus.vector_db
+    assert returned_corpus.backend_config == expected_corpus.backend_config
+    assert returned_corpus.vertex_ai_search_config == expected_corpus.vertex_ai_search_config
+    assert returned_corpus.corpus_type_config == expected_corpus.corpus_type_config
 
 
 def rag_file_eq(returned_file, expected_file):