From 8c7b1b6d9e06e9a9abb44296554b853808a046cf Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 11 Sep 2025 21:16:23 +0200 Subject: [PATCH] fix: Resolve issue with accessing dataset name --- cognee/tests/test_permissions.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cognee/tests/test_permissions.py b/cognee/tests/test_permissions.py index cfa3aade2..95f769263 100644 --- a/cognee/tests/test_permissions.py +++ b/cognee/tests/test_permissions.py @@ -79,7 +79,7 @@ async def main(): print("\n\nExtracted sentences are:\n") for result in search_results: print(f"{result}\n") - assert search_results[0].dataset_name == "NLP", ( + assert search_results[0]["dataset_name"] == "NLP", ( f"Dict must contain dataset name 'NLP': {search_results[0]}" ) @@ -93,7 +93,7 @@ async def main(): print("\n\nExtracted sentences are:\n") for result in search_results: print(f"{result}\n") - assert search_results[0].dataset_name == "QUANTUM", ( + assert search_results[0]["dataset_name"] == "QUANTUM", ( f"Dict must contain dataset name 'QUANTUM': {search_results[0]}" ) @@ -170,7 +170,7 @@ async def main(): for result in search_results: print(f"{result}\n") - assert search_results[0].dataset_name == "QUANTUM", ( + assert search_results[0]["dataset_name"] == "QUANTUM", ( f"Dict must contain dataset name 'QUANTUM': {search_results[0]}" )