fix: lint errors

This commit is contained in:
Boris Arzentar 2025-11-04 14:51:14 +01:00
parent ba4e5c3c69
commit 01a4430be1
No known key found for this signature in database
GPG key ID: D5CC274C784807B7
2 changed files with 2 additions and 5 deletions

View file

@ -13,4 +13,4 @@ async def has_data_related_nodes(dataset_id: UUID, data_id: UUID, session: Async
)
data_related_node = await session.scalar(query_statement)
return data_related_node != None
return data_related_node is not None

View file

@ -288,10 +288,7 @@ async def main(mock_create_structured_output: AsyncMock):
)
johns_data_id = add_john_result.data_ingestion_info[0]["data_id"]
add_marie_result = await cognee.add(
"Marie works for Apple as well. She is a software engineer on MacOS project."
)
# maries_data_id = add_marie_result.data_ingestion_info[0]["data_id"]
await cognee.add("Marie works for Apple as well. She is a software engineer on MacOS project.")
cognify_result: dict = await cognee.cognify()
dataset_id = list(cognify_result.keys())[0]