fix tests

langchain-ai · Oct 10, 2024 · 39f9c8c · 39f9c8c
1 parent 789562b
commit 39f9c8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/astradb/tests/integration_tests/test_graphvectorstore.py b/libs/astradb/tests/integration_tests/test_graphvectorstore.py
@@ -353,9 +353,9 @@ def test_gvs_add_nodes(
         assert hits[0].page_content == "[0, 2]"
         md0 = hits[0].metadata
         assert md0["m"] == 0
-        assert any(isinstance(v, list) for k, v in md0.items() if k != "m")
+        assert any(isinstance(v, set) for k, v in md0.items() if k != "m")
         assert hits[1].id != "id0"
         assert hits[1].page_content == "[0, 1]"
         md1 = hits[1].metadata
         assert md1["m"] == 1
-        assert any(isinstance(v, list) for k, v in md1.items() if k != "m")
+        assert any(isinstance(v, set) for k, v in md1.items() if k != "m")