From 9a9eacf8c893d8db04d6a44a181400cb798c9639 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Fri, 16 Jan 2026 14:11:06 +0100 Subject: [PATCH] refactor: update test with cache change --- cognee/tests/test_permissions.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cognee/tests/test_permissions.py b/cognee/tests/test_permissions.py index 10696441e..9d949c92b 100644 --- a/cognee/tests/test_permissions.py +++ b/cognee/tests/test_permissions.py @@ -41,14 +41,14 @@ async def _reset_engines_and_prune() -> None: except Exception: pass - from cognee.infrastructure.databases.graph.get_graph_engine import create_graph_engine from cognee.infrastructure.databases.relational.create_relational_engine import ( create_relational_engine, ) - from cognee.infrastructure.databases.vector.create_vector_engine import create_vector_engine + from cognee.infrastructure.databases.vector.create_vector_engine import _create_vector_engine + from cognee.infrastructure.databases.graph.get_graph_engine import _create_graph_engine - create_graph_engine.cache_clear() - create_vector_engine.cache_clear() + _create_graph_engine.cache_clear() + _create_vector_engine.cache_clear() create_relational_engine.cache_clear() await cognee.prune.prune_data()