fix lint

pinecone-io · Oct 4, 2023 · 799f3ac · 799f3ac
1 parent 2c9a30e
commit 799f3ac
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/resin/knoweldge_base/knowledge_base.py b/resin/knoweldge_base/knowledge_base.py
@@ -250,7 +250,9 @@ def query(self,
         results = self._reranker.rerank(results)
 
         return [
-            QueryResult(**r.dict(exclude={'values', 'sprase_values'})) for r in results
+            QueryResult(
+                **r.dict(exclude={'values', 'sprase_values', 'document_id'})
+            ) for r in results
         ]
 
     def _query_index(self,

diff --git a/tests/system/knowledge_base/test_knowledge_base.py b/tests/system/knowledge_base/test_knowledge_base.py
@@ -232,7 +232,7 @@ def test_upsert_dataframe_with_wrong_schema(knowledge_base, documents):
     df = pd.DataFrame([{"id": doc.id, "txt": doc.text, "metadata": doc.metadata}
                        for doc in documents])
 
-    with pytest.raises(ValidationError) as e:
+    with pytest.raises(ValidationError):
         knowledge_base.upsert_dataframe(df)
 
 
@@ -241,7 +241,7 @@ def test_upsert_dataframe_with_redundant_col(knowledge_base, documents):
                         "bla": "bla"}
                        for doc in documents])
 
-    with pytest.raises(ValidationError) as e:
+    with pytest.raises(ValidationError):
         knowledge_base.upsert_dataframe(df)