gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: lint errors

Browse source
This commit is contained in:
Boris Arzentar 2025-04-30 15:43:10 +02:00
parent 44cc74994f
commit c5d5c5578a
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/databases/graph/neo4j_driver/deadlock_retry.py
View file

@ -53,7 +53,7 @@ def deadlock_retry(max_retries=5):
await wait()
else:
raise # Re-raise the original error
except DatabaseUnavailable as error:
except DatabaseUnavailable:
if attempt >= max_retries:
raise # Re-raise the original error

4
cognee/tests/unit/infrastructure/databases/graph/neo4j_deadlock_test.py
View file

@ -26,7 +26,7 @@ async def test_deadlock_retry():
wrapped_function = deadlock_retry(max_retries=2)(mock_function)
result = await wrapped_function(self=None)
assert result == True, "Function should have succeded on second time"
assert result, "Function should have succeded on second time"
async def test_deadlock_retry_exhaustive():
@ -39,7 +39,7 @@ async def test_deadlock_retry_exhaustive():
wrapped_function = deadlock_retry(max_retries=2)(mock_function)
result = await wrapped_function(self=None)
assert result == True, "Function should have succeded on second time"
assert result, "Function should have succeded on second time"
if __name__ == "__main__":